arreglo8

app.py

@@ -3,17 +3,13 @@ from fastai.text.all import *
 from huggingface_hub import from_pretrained_fastai
 
 repo_id = "almangad/entregable3" 
-
 learner = from_pretrained_fastai(repo_id)
-
-labels = list(learner.dls.vocab)
+labels = learner.dls.vocab
 
 def predict_text(texto):
-    res = learner.predict(texto)
-    
-    probs = res[-1] 
-    
-    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+    pred_class, pred_idx, probs = learner.predict(texto)
+    return {str(learner.dls.vocab[i]): float(probs[i]) for i in range(len(probs))}
+
 demo = gr.Interface(
     fn=predict_text, 
     inputs=gr.Textbox(lines=3, placeholder="Text here..."),
@@ -24,5 +20,4 @@ demo = gr.Interface(
     ],
     cache_examples=False
 )
-
 demo.launch()