Update app.py

app.py

@@ -72,7 +72,7 @@ print(f"Checkpoint path: {ckpt_path}")
 # ---------------------------------------------------------------------------
 # 3. 实例化 & 加载权重
 model = SwinClassifier(MODEL_NAME, NUM_CLASSES, pretrained=False).to(device)
-state = torch.load(ckpt_path, map_location=device)
+state = torch.load(ckpt_path, map_location=device, weights_only=False)
 model.load_state_dict(state.get("model_state_dict", state), strict=True)
 model.eval()
 print("✅  Model loaded.")