Update app.py

app.py

@@ -1,5 +1,6 @@
 import torch
 import gradio as gr
+import json
 from transformers import pipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -8,7 +9,9 @@ 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 {"predictions": preds}
+    prediction_json = json.dumps(preds)
+    prediction_dict = json.loads(prediction_json)
+    return prediction_dict
 
 description = """
 """
@@ -18,7 +21,7 @@ gr.Interface(
     inputs=[
         gr.inputs.Image(label="Image to classify", type="pil"),
     ],
-    outputs=gr.outputs.JSON(),
+    outputs="label",
     title="Comparateur d'image",
     description=description
 ).launch()