Hugging Face's logo

Spaces:
Ahmedhassan54
/
Image-Classification
Running

App Files Community
Ahmedhassan54 commited on
Commit
cd90460
·
verified ·
1 Parent(s): 4bafa27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ logging.basicConfig(level=logging.INFO)
15
  logger = logging.getLogger(__name__)
16
 
17
  # Configuration
18
- MODEL_REPO = "Ahmedhassan54/Image-classifier"
19
- MODEL_FILE = "final_model.h5"
20
 
21
  # Initialize model
22
  model = None
 
15
  logger = logging.getLogger(__name__)
16
 
17
  # Configuration
18
+ MODEL_REPO = "Ahmedhassan54/Image-Classification-Model"
19
+ MODEL_FILE = "best_model.keras"
20
 
21
  # Initialize model
22
  model = None