Update app.py

app.py

@@ -159,13 +159,9 @@ def run_fhe(user_id):
     query["evaluation_key"] = encoded_evaluation_key
     query["encrypted_encoding"] = encrypted_quantized_encoding
     headers = {"Content-type": "application/json"}
-    if "legal_rating" in model_names:
-            response = requests.post(
-            "http://localhost:8000/predict_legal", data=json.dumps(query), headers=headers
-        )
-    else:
-        response = requests.post(
-                "http://127.0.0.1:8000/predict_sentiment", data=json.dumps(query), headers=headers
+
+    response = requests.post(
+                ""http://localhost:8000/predict_sentiment", data=json.dumps(query), headers=headers
             )
     encrypted_prediction = base64.b64decode(response.json()["encrypted_prediction"])