Update app.py

app.py

@@ -13,13 +13,29 @@ def predict(image):
 description = """
 """
 
-gr.Interface(
+def format_output(output):
+    # Format the JSON output for better display
+    formatted_output = ""
+    for i, pred in enumerate(output):
+        formatted_output += f"{i}: {{\n"
+        formatted_output += f"  score: {pred['score']},\n"
+        formatted_output += f"  label: {pred['label']}\n"
+        formatted_output += "}\n"
+    return formatted_output
+
+# Create the Gradio interface
+iface = gr.Interface(
     fn=predict,
     inputs=[
-        gr.components.Image(label="Image to classify", type="pil"),
-        # gr.inputs.Textbox(lines=1, label="Comma separated candidate labels", placeholder="Enter labels separated by ', '",)
+        gr.inputs.Image(label="Image to classify", type="pil"),
     ],
     outputs=gr.outputs.JSON(),
-    title="Comparateur d'image",
+    title="Image Classifier",
     description=description
-).launch()
+)
+
+# Set the post-processing hook to format the output
+iface.interface_output_postprocessing = format_output
+
+# Launch the interface
+iface.launch()