Update app.py

app.py

@@ -223,14 +223,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://localhost: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"])
 
     # Save encrypted_prediction in a file, since too large to pass through regular Gradio