ehtyalee commited on
Commit
69fd3e6
·
verified ·
1 Parent(s): e71e57a

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ import numpy as np
14
  # --- Configuration ---
15
  # REMOVED: pkl_file_path = '/content/finetune_vit_model.pkl'
16
  # ADDED: Path to the directory created by unzipping
17
- hf_model_directory = '.best-model-hf'
18
  # IMPORTANT: This MUST match the base model used for fine-tuning to load the correct feature extractor
19
  model_checkpoint = "google/vit-base-patch16-224"
20
  device_to_use = device('cuda' if cuda.is_available() else 'cpu')
 
14
  # --- Configuration ---
15
  # REMOVED: pkl_file_path = '/content/finetune_vit_model.pkl'
16
  # ADDED: Path to the directory created by unzipping
17
+ hf_model_directory = 'best-model-hf'
18
  # IMPORTANT: This MUST match the base model used for fine-tuning to load the correct feature extractor
19
  model_checkpoint = "google/vit-base-patch16-224"
20
  device_to_use = device('cuda' if cuda.is_available() else 'cpu')