FrAnKu34t23 commited on
Commit
8572cb2
·
verified ·
1 Parent(s): eab1518

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', os.path.join('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')