fix!: set weights_only to false

app.py

@@ -39,7 +39,7 @@ if not os.path.exists(model_path):
 
 # Load the trained model
 model = ViTForImageClassification.from_pretrained(
-    model_path, config=config, ignore_mismatched_sizes=True
+    model_path, config=config, ignore_mismatched_sizes=True, weights_only=False
 )
 model.classifier = nn.Linear(model.config.hidden_size, 2)
 model.to(device)