Update gradio_app.py

gradio_app.py

@@ -28,6 +28,7 @@ from huggingface_hub import snapshot_download
 import os
 
 
+
 def download_model():
     REPO_ID = 'Doubiiu/ToonCrafter'
     filename_list = ['model.ckpt']
@@ -82,8 +83,20 @@ def extract_frames(video_path):
     return frame_list
 
 
-# class Image2Video():    
-    # def __init__(self,result_dir='./tmp/',gpu_num=1,resolution='256_256') -> None:
+# 设置仓库地址和要下载的文件名
+repo_id = "Doubiiu/ToonCrafter"
+filename = "sketch_encoder.ckpt"
+
+# 指定下载的目标目录
+target_directory = "./control_models/"
+
+# 确保目标目录存在，如果不存在则创建它
+os.makedirs(target_directory, exist_ok=True)
+
+# 下载文件到本地，并指定保存路径
+local_dir = snapshot_download(repo_id=repo_id, allow_patterns=[filename], local_dir=target_directory, local_dir_use_symlinks=False)
+
+print(f"文件已下载到: {local_dir}")
 
 resolution = '320_512'
 resolution = (320, 512)
@@ -253,20 +266,7 @@ i2v_examples_interp_512 = [
 ]
 
 
-# 设置仓库地址和要下载的文件名
-repo_id = "Doubiiu/ToonCrafter"
-filename = "sketch_encoder.ckpt"
-
-# 指定下载的目标目录
-target_directory = "./control_models/"
-
-# 确保目标目录存在，如果不存在则创建它
-os.makedirs(target_directory, exist_ok=True)
-
-# 下载文件到本地，并指定保存路径
-local_dir = snapshot_download(repo_id=repo_id, allow_patterns=[filename], local_dir=target_directory, local_dir_use_symlinks=False)
 
-print(f"文件已下载到: {local_dir}")
 
 
 def dynamicrafter_demo(result_dir='./tmp/', res=512):