Upload 3 files

README.md

@@ -1,12 +1,9 @@
----
-title: Waste Classifier
-emoji: 📚
-colorFrom: blue
-colorTo: indigo
-sdk: gradio
-sdk_version: 5.34.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# ♻️ Live Waste Classifier
+
+This app uses a Hugging Face pretrained model (`prithivMLmods/Recycling-Net-11`) to classify real-time webcam images into **Recyclable** and **Non-Recyclable** categories.
+
+## Model Labels
+- **Recyclable**: cardboard, glass, metal, paper, plastic, can, carton
+- **Non-Recyclable**: food waste, trash, garbage, organic
+
+Made with 🤗 Hugging Face + Gradio

app.py

@@ -0,0 +1,40 @@
+from transformers import AutoImageProcessor, SiglipForImageClassification
+from PIL import Image
+import torch
+import gradio as gr
+
+model_name = "prithivMLmods/Recycling-Net-11"
+processor = AutoImageProcessor.from_pretrained(model_name)
+model = SiglipForImageClassification.from_pretrained(model_name)
+
+recyclable_labels = ["cardboard", "glass", "metal", "paper", "plastic", "can", "carton"]
+non_recyclable_labels = ["food waste", "trash", "garbage", "organic"]
+
+id2label = model.config.id2label
+
+def classify_webcam(image):
+    if image is None:
+        return "No Image", None
+
+    inputs = processor(images=image, return_tensors="pt")
+    with torch.no_grad():
+        logits = model(**inputs).logits
+        probs = torch.nn.functional.softmax(logits, dim=1).squeeze().tolist()
+
+    pred_idx = max(range(len(probs)), key=lambda i: probs[i])
+    pred_label = id2label[pred_idx].lower()
+    label_type = "Recyclable" if any(w in pred_label for w in recyclable_labels) else "Non-Recyclable"
+    confidence = f"{probs[pred_idx]*100:.1f}%"
+    return f"{label_type} ({pred_label}, {confidence})"
+
+demo = gr.Interface(
+    fn=classify_webcam,
+    inputs=gr.Image(source="webcam", tool="editor", streaming=False),
+    outputs="text",
+    live=True,
+    title="♻️ Live Waste Classifier",
+    description="Classifies waste items as Recyclable or Non-Recyclable using a Hugging Face model."
+)
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,4 @@
+transformers
+torch
+Pillow
+gradio