danielhshi8224 commited on
Commit
823e041
·
1 Parent(s): 0aba71b
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -12,15 +12,15 @@ MODEL_ID = "dshi01/my-benthic-classifier"
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)
17
- if os.path.exists(test_path):
18
- model_path = test_path
19
- print(f"✓ Found model: {name}")
20
- break
21
 
22
- if model_path is None:
23
- raise FileNotFoundError(f"Could not find model file. Tried: {possible_names}")
24
 
25
  # Species categories (7 classes)
26
  SPECIES_CATEGORIES = [
 
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)
17
+ # if os.path.exists(test_path):
18
+ # model_path = test_path
19
+ # print(f"✓ Found model: {name}")
20
+ # break
21
 
22
+ # if model_path is None:
23
+ # raise FileNotFoundError(f"Could not find model file. Tried: {possible_names}")
24
 
25
  # Species categories (7 classes)
26
  SPECIES_CATEGORIES = [