danielhshi8224 commited on
Commit
0aba71b
·
1 Parent(s): f11ecc4

added full HF model repo for CNN

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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)