Spaces:

chitang1
/

Garbage_Classification

Sleeping

chitang1 commited on Apr 23, 2025

Commit

b15374a

verified ·

1 Parent(s): 043d714

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,8 @@ from fastai.vision.all import *
 # def label_func(filepath):
 #     return filepath.parent.name
-LEARN = load_learner('Garbage_Classification.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):