Update app.py

app.py

@@ -1,15 +1,10 @@
 import random
-import re
 import gradio as gr
 import numpy as np
 import spaces
 import torch
 from diffusers import AutoencoderKL
 from mixture_tiling_sdxl import StableDiffusionXLTilingPipeline
-from transformers import pipeline
-
-# 한글 입력 시 영어로 번역하기 위한 파이프라인 (CPU에서 동작)
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
 
 MAX_SEED = np.iinfo(np.int32).max
 SCHEDULERS = [
@@ -42,14 +37,6 @@ pipe.enable_vae_tiling()
 pipe.enable_vae_slicing()
 
 #region helper functions
-def translate_if_needed(text: str) -> str:
-    """텍스트에 한글이 포함된 경우 영어로 번역하여 반환."""
-    if re.search(r'[\uac00-\ud7a3]', text):
-        # 번역 결과는 리스트의 첫번째 요소의 "translation_text" 키에 있음
-        translated = translator(text)[0]["translation_text"]
-        return translated
-    return text
-
 def select_scheduler(scheduler_name):
     scheduler_parts = scheduler_name.split("-")
     scheduler_class_name = scheduler_parts[0]
@@ -76,11 +63,6 @@ def predict(left_prompt, center_prompt, right_prompt, negative_prompt, left_gs,
     pipe.scheduler = select_scheduler(scheduler)
     generator = torch.Generator("cuda").manual_seed(generation_seed)
     
-    # 한글 프롬프트가 입력된 경우 영어로 번역
-    left_prompt = translate_if_needed(left_prompt)
-    center_prompt = translate_if_needed(center_prompt)
-    right_prompt = translate_if_needed(right_prompt)
-    
     target_height = int(target_height)
     target_width = int(target_width)
     tile_height = int(tile_height)
@@ -240,7 +222,7 @@ button:hover, .btn:hover {
 title = """
 <h1 align="center" style="margin-bottom: 0.2em;"> 🤗 Panorama X3 Image </h1>
 <p align="center" style="font-size:1.1em; color:#555;">
-    Left/Center/Right prompts (English/Korean allowed)
+    Create panoramic images with left/center/right prompts
 </p>
 """
 
@@ -254,15 +236,15 @@ with gr.Blocks(css=css, title="SDXL Tiling Pipeline") as app:
             with gr.Row():
                 with gr.Column(variant="panel"):
                     gr.Markdown("### Left Region")
-                    left_prompt = gr.Textbox(lines=4, placeholder="e.g., 울창한 숲과 햇살이 비추는 나무...", label="Left Prompt (English/Korean allowed)")
+                    left_prompt = gr.Textbox(lines=4, placeholder="e.g., Lush green forest with sun rays...", label="Left Prompt")
                     left_gs = gr.Slider(minimum=0, maximum=15, value=7, step=1, label="Left CFG Scale")
                 with gr.Column(variant="panel"):
                     gr.Markdown("### Center Region")
-                    center_prompt = gr.Textbox(lines=4, placeholder="e.g., 잔잔한 호수와 반짝이는 수면...", label="Center Prompt (English/Korean allowed)")
+                    center_prompt = gr.Textbox(lines=4, placeholder="e.g., Crystal clear lake reflecting...", label="Center Prompt")
                     center_gs = gr.Slider(minimum=0, maximum=15, value=7, step=1, label="Center CFG Scale")
                 with gr.Column(variant="panel"):
                     gr.Markdown("### Right Region")
-                    right_prompt = gr.Textbox(lines=4, placeholder="e.g., 웅장한 산맥과 하늘을 가르는 구름...", label="Right Prompt (English/Korean allowed)")
+                    right_prompt = gr.Textbox(lines=4, placeholder="e.g., Majestic mountains with snowy peaks...", label="Right Prompt")
                     right_gs = gr.Slider(minimum=0, maximum=15, value=7, step=1, label="Right CFG Scale")
             with gr.Row():
                 negative_prompt = gr.Textbox(
@@ -377,4 +359,4 @@ with gr.Blocks(css=css, title="SDXL Tiling Pipeline") as app:
         outputs=result,
     )
 
-app.launch(share=False)
+app.launch(share=False)