Update app.py

app.py

@@ -23,7 +23,12 @@ st.set_page_config(
 @st.cache_resource
 def load_models():
     # Vision model
-    damage_model = ViTForImageClassification.from_pretrained("microsoft/vit-base-patch16-224")
+    from huggingface_hub import login
+    login(token="HUGGINGFACE_TOKEN")
+    damage_model = ViTForImageClassification.from_pretrained(
+        "microsoft/vit-base-patch16-224", 
+        use_auth_token=True
+    )
     processor = ViTImageProcessor.from_pretrained("microsoft/vit-base-patch16-224")
     
     # Text model