Update app.py for Docker

app.py

@@ -1,16 +1,16 @@
-import gradio as gr
+from fastapi import FastAPI
 from transformers import pipeline
 
-# Cargar el modelo preentrenado DistilBERT
-pipe = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
+app = FastAPI()
 
-# Función para realizar predicción
-def predict(text):
-    result = pipe(text)
-    return result[0]['label'], result[0]['score']
+# Cargar el modelo DistilBERT para clasificación de texto
+classifier = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
 
-# Interfaz de usuario de Gradio
-interface = gr.Interface(fn=predict, inputs="text", outputs=["text", "text"])
+@app.get("/")
+def greet_json():
+    return {"Hello": "World!"}
 
-# Correr la interfaz
-interface.launch()
+@app.post("/predict/")
+def predict(text: str):
+    result = classifier(text)
+    return {"label": result[0]['label'], "score": result[0]['score']}