File size: 650 Bytes
ca0ba19
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import gradio as gr
from PIL import Image
import random

# -------------------------------
# Classification logic
# -------------------------------
def classify_item(image, description):
    categories = ["Recyclable", "Compostable", "Trash", "Harmful"]

    if description:
        desc = description.lower()

        # Compostable: fruits & vegetables
        if "banana" in desc or "apple" in desc or "fruit" in desc or "vegetable" in desc or "food" in desc or "peel" in desc or "leaf" in desc:
            category = "Compostable"

        # Harmful: medical + batteries
        elif "syringe" in desc or "needle" in desc or "battery" in desc or