Update app.py

app.py

@@ -33,7 +33,8 @@ 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)
-state_dict = torch.load(os.path.join(config["load_weights_dir"], 'model.pth'), map_location="cpu")
+hf_weight = hf_hub_download(repo_id=f"ZidongC/PanDA", filename=f"model.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()):
     model = nn.DataParallel(model)