Update app.py

app.py

@@ -2,20 +2,16 @@ from fastai.text.all import *
 import gradio as gr
 from huggingface_hub import from_pretrained_fastai
 
-# Cargar el modelo
 repo_id = "mahernto/imdbmodel"
 learner = from_pretrained_fastai(repo_id)
-labels = learner.dls.vocab
-# Función de predicción
+
 def predict(text):
-    pred = learner.predict(text)
-    if (str(pred) == "0"):
+    pred,pred_idx,probs = learner.predict(text)
+    if (pred == "0"):
         return "Negative"
-    elif (str(pred) == "1"):
+    elif (pred == "1"):
         return "Positive"
     else:
         return "Error"
 
-# Interfaz Gradio
-demo = gr.Interface(fn=predict, inputs=gr.Textbox(lines=4, placeholder="Write your film review..."), outputs="text",examples=["An outstanding film with powerful performances and a moving story. A must-watch.","Terrible acting and a weak script. Not worth watching."])
-demo.launch(share=False)
+gr.Interface(fn=predict, inputs=gr.Textbox(lines=4, placeholder="Write your film review..."), outputs="text",examples=["An outstanding film with powerful performances and a moving story. A must-watch.","Terrible acting and a weak script. Not worth watching."]).launch(share=False)