Update app.py

app.py

@@ -1,4 +1,4 @@
- import gradio as gr
+import gradio as gr
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForTokenClassification
@@ -40,18 +40,6 @@ async def predict(input: TextInput):
     
     return {"entities": entities}
 
-# Iniciar el servidor de FastAPI en un hilo separado
-def start_api():
-    uvicorn.run(app, host="0.0.0.0", port=8000)
-
-api_thread = Thread(target=start_api, daemon=True)
-api_thread.start()
-
-# Configurar Gradio
-def predict_gradio(text):
-    response = requests.post("http://localhost:8000/predict", json={"text": text})
-    entities = response.json().get("entities", [])
-    return entities
-
-demo = gr.Interface(fn=predict_gradio, inputs="text", outputs="json")
-demo.launch(share=True)
+# Iniciar el servidor de FastAPI
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=8000)