Update app.py

app.py

@@ -8,21 +8,16 @@ learn = from_pretrained_fastai("ivferns/clasificador_textos")
 # Función de predicción
 def clasificar_texto(texto):
     pred, idx, probs = learn.predict(texto)
-    # Obtener top-3 probabilidades y sus índices
-    top_vals, top_idxs = probs.topk(3)
-    top_vals = top_vals.tolist()
-    top_idxs = top_idxs.tolist()
-    # Construir un diccionario {clase: probabilidad}
-    return { CLASSES[i]: float(top_vals[j]) 
-             for j, i in enumerate(top_idxs) }
+    return str(pred)
     
 # Interfaz Gradio
 iface = gr.Interface(
     fn=clasificar_texto,
     inputs=gr.Textbox(lines=5, label="Introduce el texto"),
-    outputs=gr.Label(num_top_classes=3, label="Top 3 predicciones"),
+    outputs=gr.Label(label="Prediccion"),
     title="Clasificador de textos",
     description="Modelo de clasificación de textos fine-tuned con FastAI y desplegado en Hugging Face Spaces."
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()