Update app.py

app.py

@@ -42,6 +42,7 @@ MODEL_NAME = 'G_240000.pth'
 CHECKPOINT_REPO = 'WitchHuntTV/checkpoint_best_legacy_500.pt'
 model_pth = hf_hub_download(repo_id=MODEL_REPO,filename=MODEL_NAME)
 model_configfile = hf_hub_download(repo_id=MODEL_REPO,filename='config.json')
+checkpoint_pt = hf_hub_download(repo_id=CHECKPOINT_REPO,filename='checkpoint_best_legacy_500.pt',local_dir='./pretrain')
 class HParams():
     def __init__(self, **kwargs):
         for k, v in kwargs.items():