Create app.py

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+from setfit import SetFitModel
+
+# Cargar el modelo
+model = SetFitModel.from_pretrained("desarrolloasesoreslocales/SetFitPruebaRecorte")
+
+# Mapear las etiquetas
+
+# Definir la función de predicción
+def predict(payload):
+    results = model.predict_proba([payload])[0]
+    
+    # return "\n".join([f"Label: {id2label[i]},      Score: {round(result.item()*100, 2)} %" for i, result in enumerate(results)])
+    return results
+
+# Crear una interfaz Gradio
+iface = gr.Interface(
+    fn=predict, 
+    inputs=gr.inputs.Textbox(), 
+    outputs=gr.outputs.Textbox(), 
+    live=False,
+    layout='vertical',
+    title="Clasificador de Texto"
+)
+
+# Iniciar la interfaz Gradio
+iface.launch()