Spaces:
Sleeping
Sleeping
Upload app.py
Browse files
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 = '
|
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')
|