Update app.py

app.py

@@ -32,7 +32,7 @@ DEVICE = 'cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is
 config = 'panda_large.yaml'
 with open(config, 'r') as f:
     config = yaml.load(f, Loader=yaml.FullLoader)
-hf_weight = hf_hub_download(repo_id=f"ZidongC/PanDA", filename=f"model.pth", repo_type="model")
+hf_weight = hf_hub_download(repo_id=f"ZidongC/PanDA", filename=f"panda_large.pth", repo_type="model")
 state_dict = torch.load(hf_weight, map_location="cpu")
 model = make(config['model'])
 if any(key.startswith('module') for key in state_dict.keys()):