Update app.py

app.py

@@ -1,6 +1,6 @@
 from huggingface_hub import from_pretrained_fastai
 import gradio as gr
-from fastai.text.all import *
+from fastai.vision.all import *
 
 # repo_id = "YOUR_USERNAME/YOUR_LEARNER_NAME"
 repo_id = "allopeap/emociones"
@@ -9,8 +9,8 @@ learner = from_pretrained_fastai(repo_id)
 labels = learner.dls.vocab
 
 # Definimos una función que se encarga de llevar a cabo las predicciones
-def predict(txt):
-    pred,_, probs = learner.predict(txt)
+def predict(text):
+    pred,pred_idx, probs = learner.predict(text)
     if pred == '0':
         return "Positive"
     elif pred == 1: