Update app.py

app.py

@@ -1,25 +1,14 @@
-from huggingface_hub import from_pretrained_fastai
-import gradio as gr
 from fastai.text.all import *
+import gradio as gr
 
-repo_id = "mahernto/imdbmodel"
-learner = from_pretrained_fastai(repo_id)
-
-labels = learner.dls.vocab 
+# Cargar el modelo
+learn = load_learner('model.pkl')
 
-# Función para hacer predicciones con texto
+# Función de predicción
 def predict(text):
     pred, _, probs = learn.predict(text)
-    labels = learn.dls.vocab
-    return {label: float(prob) for label, prob in zip(labels, probs)}
+    return f"Predicción: {pred} (confianza: {probs.max():.2%})"
 
-# Interfaz Gradio para texto y etiquetas de salida
-gr.Interface(
-    fn=predict,
-    inputs=gr.Textbox(lines=2, placeholder="Write your movie review here..."),
-    outputs=gr.Label(num_top_classes=2),
-    examples=[
-        ["This movie was fantastic, I loved the plot and acting!"],
-        ["The film was boring and too long."]
-    ]
-).launch(share=False)
+# Interfaz Gradio
+demo = gr.Interface(fn=predict, inputs=gr.Textbox(lines=4, placeholder="Escribe un texto..."), outputs="text")
+demo.launch()