Upload 6 files

.gitattributes

@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+glass.png filter=lfs diff=lfs merge=lfs -text
+plastic[[:space:]]copy.jpg filter=lfs diff=lfs merge=lfs -text
+plastic.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,25 +1,25 @@
-from fastai.vision.all import *
-import gradio as gr
-
-# Load the exported model
-learn = load_learner('trash_model(1s).pkl')
-
-# Define labels (make sure they match your model's training labels)
-labels = learn.dls.vocab
-
-# Define prediction function
-def classify_trash(img):
-    pred_class, pred_idx, probs = learn.predict(img)
-    return {labels[i]: float(probs[i]) for i in range(len(labels))}
-
-# Gradio Interface
-interface = gr.Interface(
-    fn=classify_trash,
-    inputs=gr.Image(type="pil"),
-    outputs=gr.Label(num_top_classes=5),
-    title="Trash Classifier",
-    description="Upload a trash image to classify it into one of 5 categories."
-)
-
-# Launch app
-interface.launch()
+from fastai.vision.all import *
+import gradio as gr
+
+# Load the exported model
+learn = load_learner('trash_model(1s).pkl')
+
+# Define labels (make sure they match your model's training labels)
+labels = learn.dls.vocab
+
+# Define prediction function
+def classify_trash(img):
+    pred_class, pred_idx, probs = learn.predict(img)
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+
+# Gradio Interface
+interface = gr.Interface(
+    fn=classify_trash,
+    inputs=gr.Image(type="pil"),
+    outputs=gr.Label(num_top_classes=5),
+    title="Trash Classifier",
+    description="Upload a trash image to classify it into one of 5 categories."
+)
+
+# Launch app
+interface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+fastai
+gradio

trash_model(1).pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c79e9f811d1267483d240511221d32b603b27a5897c183d2e1f5273e9c39ea4
+size 114704730