Spaces:
Sleeping
Sleeping
danielhshi8224 commited on
Commit ·
0aba71b
1
Parent(s): f11ecc4
added full HF model repo for CNN
Browse files
app.py
CHANGED
|
@@ -9,8 +9,8 @@ BASE_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
| 9 |
MODEL_ID = "dshi01/my-benthic-classifier"
|
| 10 |
|
| 11 |
# Try different possible filenames
|
| 12 |
-
possible_names = ['ConvNextmodel.pth', 'convnextmodel.pth', 'ConvNext_model.pth']
|
| 13 |
-
model_path = None
|
| 14 |
|
| 15 |
for name in possible_names:
|
| 16 |
test_path = os.path.join(BASE_DIR, name)
|
|
|
|
| 9 |
MODEL_ID = "dshi01/my-benthic-classifier"
|
| 10 |
|
| 11 |
# Try different possible filenames
|
| 12 |
+
# possible_names = ['ConvNextmodel.pth', 'convnextmodel.pth', 'ConvNext_model.pth']
|
| 13 |
+
# model_path = None
|
| 14 |
|
| 15 |
for name in possible_names:
|
| 16 |
test_path = os.path.join(BASE_DIR, name)
|