Uploading our food not food classifier demo from a notebook!

app.py

@@ -12,6 +12,13 @@ spanish_sentence_classification_by_school_subject_pipeline = pipeline(task="text
                                     device="cuda" if torch.cuda.is_available() else "cpu",
                                     batch_size=32)    
 
+def classify_text(text):
+    # Usa el clasificador
+    result = spanish_sentence_classification_by_school_subject_pipeline(text)
+    # Ahora accedemos al primer diccionario en la primera lista
+    return result[0][0]['label'], result[0][0]['score']
+
+
 # 3. Create a Gradio interface
 description = """
 Un clasificador de texto que indica a qué asignatura se refiere la frase. 
@@ -20,7 +27,7 @@ Fine-tuned from [DistilBERT](https://huggingface.co/distilbert/distilbert/distil
 """
 
 demo = gr.Interface(
-    fn = spanish_sentence_classification_by_school_subject_pipeline,
+    fn = classify_text,
     inputs = "text",
     outputs=gr.Label(num_top_classes=10),
     title="📚🔍 Clasificador de asignaturas",