tonicanada commited on
Commit
e4f80b4
·
verified ·
1 Parent(s): 626385f

Uploading our food not food classifier demo from a notebook!

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ food_not_food_classifier = pipeline(task="text-classification",
15
  def classify_text(text):
16
  # Usa el clasificador
17
  result = food_not_food_classifier(text)
18
- print(result) # Imprime el resultado para inspeccionarlo
19
- return result[0]['label'], result[0]['score']
20
 
21
 
22
  # 3. Create a Gradio interface
 
15
  def classify_text(text):
16
  # Usa el clasificador
17
  result = food_not_food_classifier(text)
18
+ # Ahora accedemos al primer diccionario en la primera lista
19
+ return result[0][0]['label'], result[0][0]['score']
20
 
21
 
22
  # 3. Create a Gradio interface