Fix API URL

app.py

@@ -4,28 +4,29 @@ import os
 import time
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
-API_URL = "https://api-inference.huggingface.co/models/Hate-speech-CNERG/dehatebert-mono-english"
+API_URL = "https://router.huggingface.co/hf-inference/models/Hate-speech-CNERG/dehatebert-mono-english"
 
 def query(text):
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    response = requests.post(API_URL, headers=headers, json={
-        "inputs": text,
-        "options": {"wait_for_model": True}
-    })
-    return response.status_code, response.text
+    for i in range(3):
+        response = requests.post(API_URL, headers=headers, json={
+            "inputs": text,
+            "options": {"wait_for_model": True}
+        })
+        if response.status_code == 200:
+            return response.json()
+        time.sleep(3)
+    return {"error": f"Status: {response.status_code}, Response: {response.text}"}
 
 def predict(comment):
     if not comment.strip():
         return "⚠️ Please enter some text.", ""
     
     try:
-        status, text = query(comment)
+        result = query(comment)
         
-        if status != 200:
-            return f"Status {status}: {text}", ""
-        
-        import json
-        result = json.loads(text)
+        if isinstance(result, dict) and "error" in result:
+            return f"API Error: {result['error']}", ""
         
         if isinstance(result, list):
             result = result[0][0]
@@ -41,7 +42,7 @@ def predict(comment):
             
             return prediction, f"{probability}%"
         
-        return f"Response: {text}", ""
+        return f"Unexpected: {result}", ""
     
     except Exception as e:
         return f"Exception: {str(e)}", ""