Dyno1307 commited on
Commit
cf0aaae
·
1 Parent(s): bd75eed

Fix model path to use flagship_model.keras

Browse files
Files changed (1) hide show
  1. src/app.py +1 -1
src/app.py CHANGED
@@ -22,7 +22,7 @@ with open(os.path.join(os.path.dirname(os.path.abspath(__file__)), 'labels.json'
22
  CLASSIFICATION_LABELS = json.load(f)
23
 
24
  # Get the absolute path to the classification model
25
- model_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'models/classification_model.keras')
26
  # Load the pre-trained classification model
27
  classification_model = load_model(model_path)
28
 
 
22
  CLASSIFICATION_LABELS = json.load(f)
23
 
24
  # Get the absolute path to the classification model
25
+ model_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'models/flagship_model.keras')
26
  # Load the pre-trained classification model
27
  classification_model = load_model(model_path)
28