FrAnKu34t23 commited on
Commit
c8c6d5d
·
verified ·
1 Parent(s): 26fbc52

Update app.py

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