Update app.py

app.py

@@ -89,9 +89,9 @@ if weight_path is None:
 model = MultitaskProteinModel(config)
 if weight_path.endswith(".safetensors"):
     from safetensors.torch import load_file
-    state_dict = load_file(weight_path, device="cpu")
+    state_dict = load_file(weight_path, device="cpu", weights_only=False)
 else:
-    state_dict = torch.load(weight_path, map_location="cpu")
+    state_dict = torch.load(weight_path, map_location="cpu", weights_only=False)
 missing, unexpected = model.load_state_dict(state_dict, strict=False)
 if missing or unexpected:
     print(f"[load_state_dict] missing={len(missing)} unexpected={len(unexpected)}")