Spaces:
Sleeping
Sleeping
Fix model path to use flagship_model.keras
Browse files- 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/
|
| 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 |
|