Spaces:

chitang1
/

Garbage_Classification

Sleeping

chitang1 commited on Apr 23, 2025

Commit

937edd6

verified ·

1 Parent(s): ed3888b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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