Update app.py

app.py

@@ -1,12 +1,11 @@
 import gradio as gr
-import os
 
 # Categories
 compostable = ["vegetable", "fruit", "food", "leaves"]
 recyclable = ["plastic", "paper", "glass", "metal", "bottle", "can"]
 harmful = ["syringe", "battery", "medical", "chemical"]
 
-# Bin images stored in 'bins/' folder
+# Bin images in 'bins/' folder
 bin_images = {
     "Compostable": "bins/stock-vector-bin-recycle-plastic-green-wheelie-bin-for-waste-isolated-on-white-background-green-bin-with-1306559440.jpg",
     "Recyclable": "bins/blue-recycling-bin-with-recycle-symbol-for-waste-management-and-sustainability-png.png",
@@ -22,14 +21,13 @@ def classify_waste(item):
     elif item in harmful:
         return "⚠️ Harmful/Non-Decomposable Waste", bin_images["Harmful"]
     else:
-        # Unknown items treated as compostable
+        # Unknown items treated as Compostable
         return "✅ Compostable Waste", bin_images["Compostable"]
 
-# Gradio UI
 iface = gr.Interface(
     fn=classify_waste,
     inputs="text",
-    outputs=["text", gr.Image(type="auto")],  # Make sure output type is an image
+    outputs=["text", gr.Image(type="filepath")],  # <-- FIXED here
     title="♻️ Eco-Friendly Waste Classifier",
     description="Enter a waste item (e.g., vegetable, plastic, syringe) and see which bin it belongs to!"
 )
@@ -38,3 +36,4 @@ if __name__ == "__main__":
     iface.launch()
 
 
+