Hugging Face's logo

Spaces:
Yumeng-Liu
/
trash-classifier-space
Sleeping

App Files Community
Yumeng Liu commited on
Commit
d6e48e1
·
1 Parent(s): a13a525

new model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ app = fastapi.FastAPI()
15
 
16
  model = saving.load_model("hf://Yumeng-Liu/trash-classifier")
17
 
18
- CLASSES = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
19
  THRESHOLD = 5.0e-1
20
 
21
 
 
15
 
16
  model = saving.load_model("hf://Yumeng-Liu/trash-classifier")
17
 
18
+ CLASSES = ['biological', 'cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
19
  THRESHOLD = 5.0e-1
20
 
21