Hugging Face's logo

Spaces:
Penthes
/
waste-classifier
Sleeping

App Files Community
Penthes commited on
Commit
24e4de1
·
verified ·
1 Parent(s): 84eda7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -44,10 +44,10 @@ def load_model():
44
  logger.info("=== DEBUGGING MODEL LOADING ===")
45
 
46
  model_files = [
47
- 'model/waste_classification_model.keras',
48
- 'model/waste_classification_model.h5',
49
- './model/waste_classification_model.keras',
50
- './model/waste_classification_model.h5',
51
  ]
52
 
53
  model = None
 
44
  logger.info("=== DEBUGGING MODEL LOADING ===")
45
 
46
  model_files = [
47
+ 'model/waste_model.keras',
48
+ 'model/waste_model.h5',
49
+ './model/waste_model.keras',
50
+ './model/waste_model.h5',
51
  ]
52
 
53
  model = None