arreglo

app.py

@@ -1,6 +1,6 @@
-from huggingface_hub import from_pretrained_fastai
 import gradio as gr
 from fastai.text.all import *
+from huggingface_hub import from_pretrained_fastai
 
 repo_id = "almangad/entregable3" 
 
@@ -8,13 +8,12 @@ learner = from_pretrained_fastai(repo_id)
 labels = learner.dls.vocab[1]
 
 def predict_text(texto):
-    res = learner.predict(texto)
-    probs = res[-1] 
+    pred, pred_idx, probs = learner.predict(texto)
     return {labels[i]: float(probs[i]) for i in range(len(labels))}
 
 demo = gr.Interface(
     fn=predict_text, 
-    inputs=gr.Textbox(lines=3, placeholder="Text here"),
+    inputs=gr.Textbox(lines=3, placeholder="Text here..."),
     outputs=gr.Label(num_top_classes=6),
     examples=[
         ["im feeling quite sad and sorry for myself but ill snap out of it soon"],
@@ -22,4 +21,4 @@ demo = gr.Interface(
     ]
 )
 
-demo.launch(share=False)
+demo.launch()