diff --git a/backend/server/malware_detector.py b/backend/server/malware_detector.py new file mode 100644 index 0000000..bb6ec11 --- /dev/null +++ b/backend/server/malware_detector.py @@ -0,0 +1,16 @@ +import os +from huggingface_hub import hf_hub_download +import requests + +API_URL = "https://api-inference.huggingface.co/models/elftsdmr/malware-url-detect" +headers = {"Authorization": "Bearer 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"} + +def query(payload): + response = requests.post(API_URL, headers=headers, json=payload) + return response.json() + +user_input = input("Enter URL here: ") +output = query({ + "inputs": user_input, +}) +print(output) \ No newline at end of file