Update app.py

app.py

@@ -10,7 +10,7 @@ image_pipeline = pipeline(task="image-classification", model="guillen/vit-basura
 
 def predict_image(input_img):
     predictions = image_pipeline(input_img)
-    return input_img, {p["label"]: p["score"] for p in predictions} 
+    return {p["label"]: p["score"] for p in predictions} 
 
 image_gradio_app = gr.Interface(
     fn=predict_image,