Update app.py

app.py

@@ -7,9 +7,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 def predict(image):
     classifier = pipeline(task="image-classification")
     preds = classifier(image)
-    prediction_json = json.dumps({"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds)
-    label = gr.Label(output_format="json", default_value=prediction_json)
-    return label
+    preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
+    return preds
 
 description = """
 """
@@ -18,10 +17,8 @@ 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 ', '",)
     ],
     outputs="label",
     title="Comparateur d'image",
     description=description
 ).launch()
-