Update gradio_app.py

gradio_app.py

@@ -86,7 +86,7 @@ def extract_frames(video_path):
     # def __init__(self,result_dir='./tmp/',gpu_num=1,resolution='256_256') -> None:
 
 resolution = '320_512'
-resolution = (int(resolution.split('_')[0]), int(resolution.split('_')[1])) #hw
+resolution = (320, 512)
 download_model()
 print("after download model")
 result_dir = "./results/"
@@ -94,8 +94,8 @@ if not os.path.exists(result_dir):
     os.mkdir(result_dir)
 
 #ToonCrafterModel
-ckpt_path='checkpoints/tooncrafter_'+resolution.split('_')[1]+'_interp_v1/model.ckpt'
-config_file='configs/inference_'+resolution.split('_')[1]+'_v1.0.yaml'
+ckpt_path='checkpoints/tooncrafter_512_interp_v1/model.ckpt'
+config_file='configs/inference_512_v1.0.yaml'
 config = OmegaConf.load(config_file)
 model_config = config.pop("model", OmegaConf.create())
 model_config['params']['unet_config']['params']['use_checkpoint']=False