Update app.py

app.py

@@ -11,10 +11,6 @@ from PIL import Image
 from diffusers.utils import load_image
 from diffusers.utils import export_to_video
 import random
-from transformers import pipeline
-
-# Translation model load
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
 # English menu labels
 english_labels = {
@@ -68,10 +64,9 @@ def convert_to_centered_scale(num):
         end = num // 2
     return tuple(range(start, end + 1))
 
-def translate_if_korean(text):
-    if any('\u3131' <= char <= '\u3163' or '\uac00' <= char <= '\ud7a3' for char in text):
-        return translator(text)[0]['translation_text']
-    return text
+def is_korean(text):
+    """한글 포함 여부 확인"""
+    return any('\u3131' <= char <= '\u3163' or '\uac00' <= char <= '\ud7a3' for char in text)
 
 @spaces.GPU(duration=85)
 def generate(prompt,
@@ -89,10 +84,10 @@ def generate(prompt,
              avg_diff_x=None,
              total_images=[],
              gradio_progress=gr.Progress()):
-    # Translate prompt and concepts if Korean
-    prompt = translate_if_korean(prompt)
-    concept_1 = translate_if_korean(concept_1)
-    concept_2 = translate_if_korean(concept_2)
+    # Check if there is Korean text and warn if so
+    if is_korean(prompt) or is_korean(concept_1) or is_korean(concept_2):
+        print("Korean text detected. The model will use it directly without translation.")
+    
     print(f"Prompt: {prompt}, ← {concept_2}, {concept_1} ➡️ . scale {scale}, interm steps {interm_steps}")
     slider_x = [concept_2, concept_1]
     if randomize_seed:
@@ -315,7 +310,7 @@ with gr.Blocks(css=css, title="Time Stream") as demo:
                         label=english_labels["From 1st to 2nd direction"]
                     )
     
-    # Examples: 예제 클릭 시 입력란에 값이 바로 삽입됨
+    # Examples
     gr.Examples(
         examples=examples,
         inputs=[prompt, concept_1, concept_2, x]
@@ -352,4 +347,4 @@ with gr.Blocks(css=css, title="Time Stream") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()