From fdc7c0e6c4d7bfa8d45fa8143bba0588760e4216 Mon Sep 17 00:00:00 2001
From: Akshatbist <133608309+Akshatbist@users.noreply.github.com>
Date: Sun, 31 Mar 2024 18:36:57 -0700
Subject: [PATCH] Create malware_detector.py

---
 backend/server/malware_detector.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 backend/server/malware_detector.py

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