FrAnKu34t23 commited on
Commit
b543dd0
·
verified ·
1 Parent(s): 61505b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ except Exception:
24
 
25
  # Configuration
26
  # Default to the moved fine-tuned checkpoint if present
27
- MODEL_PATH = os.environ.get('MODEL_PATH', os.path.join('results', 'fine_tune', 'best_model_finetuned.pth'))
28
  # Optional: if your HF model id is known (e.g. Emiel/cub-200-bird-classifier-swin), set HF_MODEL_ID env var
29
  HF_MODEL_ID = os.environ.get('HF_MODEL_ID', None)
30
  CLASS_NAMES_PATH = os.environ.get('CLASS_NAMES_PATH', 'class_names.json')
 
24
 
25
  # Configuration
26
  # Default to the moved fine-tuned checkpoint if present
27
+ MODEL_PATH = os.environ.get('MODEL_PATH', os.path.join('best_model_finetuned.pth'))
28
  # Optional: if your HF model id is known (e.g. Emiel/cub-200-bird-classifier-swin), set HF_MODEL_ID env var
29
  HF_MODEL_ID = os.environ.get('HF_MODEL_ID', None)
30
  CLASS_NAMES_PATH = os.environ.get('CLASS_NAMES_PATH', 'class_names.json')