Update app.py

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 import os
 import spaces
 import sys
+from transformers import pipeline
+
+
 from copy import deepcopy
 sys.path.append('./VADER-VideoCrafter/scripts/main')
 sys.path.append('./VADER-VideoCrafter/scripts')
@@ -10,6 +13,10 @@ sys.path.append('./VADER-VideoCrafter')
 
 from train_t2v_lora import main_fn, setup_model
 
+translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
+
+
+
 examples = [
     ["Fairy and Magical Flowers: A fairy tends to enchanted, glowing flowers.", 'huggingface-hps-aesthetic', 
      8, 901, 384, 512, 12.0, 25, 1.0, 24, 10],
@@ -29,12 +36,18 @@ examples = [
 
 model = setup_model()
 
-@spaces.GPU(duration=180)
 def gradio_main_fn(prompt, lora_model, lora_rank, seed, height, width, unconditional_guidance_scale, ddim_steps, ddim_eta,
                    frames, savefps):
     global model
     if model is None:
         return "Model is not loaded. Please load the model first."
+    
+    # 한글 입력 감지 및 번역
+    if any('\u3131' <= char <= '\u318E' or '\uAC00' <= char <= '\uD7A3' for char in prompt):
+        translated = translator(prompt, max_length=512)[0]['translation_text']
+        print(f"Translated prompt: {translated}")
+        prompt = translated
+
     video_path = main_fn(prompt=prompt,
                     lora_model=lora_model,
                     lora_rank=int(lora_rank),