Update app.py

app.py

@@ -8,28 +8,20 @@ def predict(image):
     classifier = pipeline(task="image-classification")
     preds = classifier(image)
     preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
-    return preds
-
-def format_output(output):
-    formatted_output = ""
-    for idx, pred in enumerate(output):
-        formatted_output += f"{idx}: Score: {pred['score']}, Label: {pred['label']}\n"
-    return formatted_output
+    label = gr.Label(output_format="json", default_value=preds)
+    return label
 
 description = """
 """
 
-iface = gr.Interface(
+gr.Interface(
     fn=predict,
     inputs=[
-        gr.inputs.Image(label="Image to classify", type="pil"),
+        gr.components.Image(label="Image to classify", type="pil"),
+        # gr.inputs.Textbox(lines=1, label="Comma separated candidate labels", placeholder="Enter labels separated by ', '",)
     ],
-    outputs=gr.outputs.JSON(),
-    title="Image Classifier",
+    outputs="label",
+    title="Comparateur d'image",
     description=description
-)
-
-# Apply custom formatting to the JSON output
-iface.output_formatter = format_output
+).launch()
 
-iface.launch()