app change

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
-# import numpy as np
-# from PIL import Image
 from transformers import pipeline
 
+# import numpy as np
+# from PIL import Image
 
 # from datasets import load_dataset
 # from huggingface_hub import hf_hub_download
@@ -36,7 +36,7 @@ classifier = pipeline("image-classification", model="yangy50/garbage-classificat
 #     return image
 
 def predict(img):
-    predictions = classifier(image)
+    predictions = classifier(img)
     return {p["label"]: p["score"] for p in predictions}
     # img = preprocess(img)
     # preds = model.predict(img)[0]