Update app.py

app.py

@@ -1,6 +1,5 @@
 import torch
 import gradio as gr
-import json
 from transformers import pipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -9,9 +8,7 @@ def predict(image):
     classifier = pipeline(task="image-classification")
     preds = classifier(image)
     preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
-    prediction_json = json.dumps(preds)
-    prediction_dict = json.loads(prediction_json)
-    return prediction_dict
+    return [pred["label"] for pred in preds]  # Return a list of labels only
 
 description = """
 """
@@ -21,7 +18,7 @@ gr.Interface(
     inputs=[
         gr.inputs.Image(label="Image to classify", type="pil"),
     ],
-    outputs="label",
-    title="Comparateur d'image",
+    outputs=gr.outputs.Label(),  # Use Label output instead of JSON
+    title="Image Classifier",
     description=description
 ).launch()