fff

app.py

@@ -20,9 +20,9 @@ def classify_recipe(description):
     logits = torch.FloatTensor(logits)
     probs = torch.sigmoid(logits)[0]
     
-    result = {class_label: float(prob) for class_label, prob in zip(recipe)}
+    result = {class_label: float(prob) for class_label, prob in zip(recipe,probs)}
     return result
 
-label = gr.outputs.Label(num_top_classes=5)
+label = gr.outputs.Label(num_top_classes=10)
 iface = gr.Interface(fn=classify_recipe, inputs="text", outputs=label)
 iface.launch(inline=False)