fix: fix model link

app.py

@@ -29,7 +29,7 @@ config = ViTConfig.from_pretrained("google/vit-base-patch16-224")
 config.num_labels = 2  # Binary classification
 
 # Download the model file
-model_url = "https://huggingface.co/spuun/yummy-paws/resolve/main/best_model.pth"
+model_url = "https://huggingface.co/spuun/yummy-paws/resolve/main/model.pth"
 model_path = "best_model.pth"
 
 if not os.path.exists(model_path):
@@ -39,7 +39,9 @@ if not os.path.exists(model_path):
 
 # Load the trained model
 model = ViTForImageClassification.from_pretrained(
-    model_path, config=config, ignore_mismatched_sizes=True, weights_only=False
+    model_path,
+    config=config,
+    ignore_mismatched_sizes=True,  # weights_only=False
 )
 model.classifier = nn.Linear(model.config.hidden_size, 2)
 model.to(device)