Update app.py

app.py

@@ -37,6 +37,9 @@ def classify_image(image, candidate_labels):
         for score, candidate_label in sorted(zip(probs, candidate_labels), key=lambda x: -x[0])
     ]
 
+    # Convert results to list of lists for Dataframe
+    results_for_df = [[res['label'], res['score']] for res in results]
+
     # Decision-making logic
     top_label = results[0]["label"]
     second_label = results[1]["label"]
@@ -58,7 +61,7 @@ def classify_image(image, candidate_labels):
     else:
         result = False
 
-    return result, top_label, results, messages
+    return result, top_label, results_for_df, messages
 
 iface = gr.Interface(
     fn=classify_image,
@@ -69,7 +72,7 @@ iface = gr.Interface(
     outputs=[
         gr.Label(label="Result"),
         gr.Textbox(label="Top Label"),
-        gr.Dataframe(label="Details"),
+        gr.Dataframe(headers=["Label", "Score"], label="Details"),
         gr.Textbox(label="Messages")
     ],
     title="General Action Classifier",