Update app.py

app.py

@@ -30,9 +30,18 @@ MAX_IMAGE_SIZE = 2048
 
 # DynamiCrafter 모델 설정
 def download_model():
-    # ... (기존 코드 유지)
-
+    REPO_ID = 'Doubiiu/DynamiCrafter_1024'
+    filename_list = ['model.ckpt']
+    if not os.path.exists('./checkpoints/dynamicrafter_1024_v1/'):
+        os.makedirs('./checkpoints/dynamicrafter_1024_v1/')
+    for filename in filename_list:
+        local_file = os.path.join('./checkpoints/dynamicrafter_1024_v1/', filename)
+        if not os.path.exists(local_file):
+            hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir='./checkpoints/dynamicrafter_1024_v1/', force_download=True)
+
+# 모델 다운로드 실행
 download_model()
+
 ckpt_path='checkpoints/dynamicrafter_1024_v1/model.ckpt'
 config_file='configs/inference_1024_v1.0.yaml'
 config = OmegaConf.load(config_file)