Spaces:

chitang1
/

Garbage_Classification

Sleeping

chitang1 commited on Apr 23, 2025

Commit

b3e9924

verified ·

1 Parent(s): b15374a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,8 @@ from fastai.vision.all import *
 # def label_func(filepath):
 #     return filepath.parent.name
-# LEARN = load_learner('Garbage_Classification.pkl')
-LEARN = load_learner('Garbage_Classification (1).pkl')
 LABELS = LEARN.dls.vocab
 def gradio_predict(img):

 # def label_func(filepath):
 #     return filepath.parent.name
+LEARN = load_learner('Garbage_Classification.pkl')
+# LEARN = load_learner('Garbage_Classification (1).pkl')
 LABELS = LEARN.dls.vocab
 def gradio_predict(img):