Update app.py

app.py

@@ -10,18 +10,16 @@ def classify_image(input_image):
     results = pipe(input_image)
     
     # Prepare the output string with all predictions
-    output_str = "Predictions:\n"
-    for result in results:
-        output_str += f"{result['label']}: {result['score']:.4f}\n"
+    confidences = {result['label']: float(result['score']) for result in results}
     
     # Return the concatenated string of predictions
-    return output_str
+    return confidences
 
 # Create a Gradio interface
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil", label="Upload an image"),
-    outputs=gr.Label(num_top_classes=5, label="Top 5 Predictions"),
+    outputs=gr.Label(num_top_classes=5),
     title="Image Classification",
     description="Classify food items in images using heisenberg3376/vit-base-food-items-v1"
 )