Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 import uvicorn
 from fastapi import FastAPI
 from gliner import GLiNER
+from starlette.responses import StreamingResponse, JSONResponse
 
 
 model = GLiNER.from_pretrained("urchade/gliner_medium-v2.1")
@@ -155,14 +156,14 @@ app = FastAPI()
 
 @app.head("/predict")
 def predict_head():
-    return {}
+    return StreamingResponse("", media_type="application/json")
 
 @app.get("/predict")
 def predict_get(text: str = "", labels: str = "", threshold: float = 0.3, nested_ner: bool = False):
     predict_response = requests.post('http://localhost:7860/call/predict', json={'data': [text, labels, threshold, nested_ner]}).json()
     if "event_id" not in predict_response:
         return predict_response
-    return json.loads(requests.get(f'http://localhost:7860/call/predict/{predict_response["event_id"]}').text.split("data: ", 1)[-1])
+    return JSONResponse(json.loads(requests.get(f'http://localhost:7860/call/predict/{predict_response["event_id"]}').text.split("data: ", 1)[-1]))
 
 if __name__ == "__main__":
     app = gr.mount_gradio_app(app, demo, path="/")