Update app.py

app.py

@@ -61,22 +61,21 @@ def call_chatgpt(prompt: str) -> str:
 
 def ai_judge(sentence1: str, sentence2: str) -> float:
 
-    HF_TOKEN = os.environ["HF_TOKEN"]
-    API_URL = "https://api-inference.huggingface.co/models/sentence-transformers/msmarco-distilbert-base-tas-b"
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    
+    API_URL = "https://laazu6ral9w37pfb.us-east-1.aws.endpoints.huggingface.cloud"
+    headers = {
+    	"Accept" : "application/json",
+    	"Content-Type": "application/json" 
+    }
+
     def helper(payload):
         response = requests.post(API_URL, headers=headers, json=payload)
         return response.json()
     
-    data = helper(
-        {
-            "inputs": {
-                "source_sentence": sentence1,
-                "sentences": [sentence2]
-            }
-        }
-    )
+    data = helper({
+        "source_sentence": sentence1,
+        "sentences": [sentence2],
+        "parameters": {}
+    })
 
     return data