Update app.py

app.py

@@ -73,7 +73,7 @@ def load_clip_models(name: str = "ViT-L/14", device='cuda'):
 
 def load_model(model_path: str, input_size=768, device: str = 'cuda', dtype=None):
     model = MLP(input_size=input_size)
-    state = torch.load(model_path, map_location=device)
+    state = torch.load(model_path, map_location=device, weights_only=False)
     model.load_state_dict(state)
     model.to(device)
     if dtype: