Upload app.py

app.py

@@ -14,7 +14,7 @@ import numpy as np
 # --- Configuration ---
 # REMOVED: pkl_file_path = '/content/finetune_vit_model.pkl'
 # ADDED: Path to the directory created by unzipping
-hf_model_directory = '.best-model-hf'
+hf_model_directory = 'best-model-hf'
 # IMPORTANT: This MUST match the base model used for fine-tuning to load the correct feature extractor
 model_checkpoint = "google/vit-base-patch16-224"
 device_to_use = device('cuda' if cuda.is_available() else 'cpu')