Update app.py

app.py

@@ -20,7 +20,7 @@ from funcs import (
     get_latent_z,
     save_videos
 )
-
+from transformers import pipeline
 
 def download_model():
     REPO_ID = 'Doubiiu/DynamiCrafter_1024'
@@ -32,7 +32,6 @@ def download_model():
         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'
@@ -45,9 +44,17 @@ model = load_model_checkpoint(model, ckpt_path)
 model.eval()
 model = model.cuda()
 
+# 번역 모델 초기화
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
 @spaces.GPU(duration=300)
 def infer(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, video_length=2):
+    # 한글 입력 감지 및 번역
+    if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
+        translated = translator(prompt, max_length=512)[0]['translation_text']
+        prompt = translated
+        print(f"Translated prompt: {prompt}")
+
     resolution = (576, 1024)
     save_fps = 8
     seed_everything(seed)
@@ -105,7 +112,7 @@ i2v_examples = [
 css = """#input_img {max-width: 1024px !important} #output_vid {max-width: 1024px; max-height: 576px}"""
 
 with gr.Blocks(analytics_enabled=False, css=css) as dynamicrafter_iface:
-    gr.Markdown("이미지로 영상 생성 테스트")
+    gr.Markdown("이미지로 영상 생성 테스트 (한글 프롬프트 지원)")
     with gr.Tab(label='ImageAnimation_576x1024'):
         with gr.Column():
             with gr.Row():