Update app.py

app.py

@@ -4,9 +4,9 @@ from transformers import pipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-def predict(image, labels):
+def predict(image):
     classifier = pipeline(task="image-classification")
-    inputs = classifier(image)
+    preds = classifier(image)
     preds = [{"score": round(pred["score"], 4), "label": pred["label"]} for pred in preds]
     return preds
 
@@ -20,7 +20,7 @@ gr.Interface(
         # gr.inputs.Textbox(lines=1, label="Comma separated candidate labels", placeholder="Enter labels separated by ', '",)
     ],
     theme="grass",
-    outputs="label",
+    outputs=gr.Textbox(),
     title="Comparateur d'image",
     description=description
-).launch()
+).launch(debug=True)