Update app.py

app.py

@@ -24,8 +24,8 @@ hf_hub_download(repo_id="Kurokabe/GANime_Kimetsu-no-yaiba_Tensorflow", filename=
 gpt_path = hf_hub_download(repo_id="Kurokabe/GANime_Kimetsu-no-yaiba_Tensorflow", filename="checkpoint", subfolder="ganime_kny_video_full") 
 
 cfg = omegaconf.OmegaConf.load(here("configs/kny_video_gpt2_large_gradio.yaml"))
-cfg["model"]["first_stage_config"]["checkpoint_path"] = vqgan_path + "/checkpoint"
-cfg["model"]["transformer_config"]["checkpoint_path"] = gpt_path + "/checkpoint"
+cfg["model"]["first_stage_config"]["checkpoint_path"] = vqgan_path
+cfg["model"]["transformer_config"]["checkpoint_path"] = gpt_path
 
 model = Net2Net(**cfg["model"], trainer_config=cfg["train"], num_replicas=1)
 model.first_stage_model.build((20, *IMAGE_SHAPE))