Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import requests
 import os
 
-# Hugging Face API Token (Set in Hugging Face Spaces)
+# Hugging Face API Token (Set this in Hugging Face Spaces)
 HF_API_TOKEN = os.getenv("HF_API_TOKEN")
 
 # Hugging Face model endpoints
@@ -17,10 +17,22 @@ def get_toxicity_prediction(text, language):
         return "Error: Model not found for the selected language"
 
     url = MODEL_ENDPOINTS[language]
-    headers = {"Authorization": f"Bearer {HF_API_TOKEN}"}
-    
-    # Send request to Hugging Face API
-    response = requests.post(url, headers=headers, json={"inputs": text, "parameters": {"return_all_scores": True}})
+    headers = {
+        "Authorization": f"Bearer {HF_API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+
+    payload = {
+        "inputs": text,
+        "parameters": {
+            "return_all_scores": True
+        },
+        "options": {
+            "wait_for_model": True  # Ensures the model loads if it's inactive
+        }
+    }
+
+    response = requests.post(url, headers=headers, json=payload)
 
     if response.status_code == 200:
         predictions = response.json()[0]  # Extract first prediction result