Update app.py

app.py

@@ -51,22 +51,19 @@ def load_gallery_images():
     
     return [(os.path.join("gallery", row[2]), f"{row[0]}: {row[1]}") for row in rows]
 
-# CPU에서 번역 모델 초기화
-@spaces.CPU
-def init_translator():
-    return pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device="cpu")
-
-# CPU에서 번역 처리
-@spaces.CPU
-def translate_text(translator, text):
+# 번역기 초기화를 GPU 함수 내부로 이동
+@spaces.GPU(duration=70)
+def translate_text_gpu(text):
+    # 번역기를 함수 내부에서 초기화
+    translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en")
     if any('\u3131' <= char <= '\u3163' or '\uac00' <= char <= '\ud7a3' for char in text):
         translated = translator(text)[0]['translation_text']
         return text, translated
     return text, text
 
 # 프롬프트 처리 함수
-def process_prompt(translator, prompt):
-    return translate_text(translator, prompt)
+def process_prompt(prompt):
+    return translate_text_gpu(prompt)
 
 KEY_JSON = os.getenv("KEY_JSON")
 with open(KEY_JSON, 'r') as f:
@@ -144,11 +141,11 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, lora_scal
             yield img
 
 @spaces.GPU(duration=70)
-def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, translator, progress=gr.Progress(track_tqdm=True)):
+def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale, progress=gr.Progress(track_tqdm=True)):
     if selected_index is None:
         raise gr.Error("진행하기 전에 LoRA를 선택해야 합니다.")
     
-    original_prompt, english_prompt = process_prompt(translator, prompt)
+    original_prompt, english_prompt = process_prompt(prompt)
     
     selected_lora = loras[selected_index]
     lora_path = selected_lora["repo"]
@@ -285,12 +282,7 @@ if not os.path.exists('gallery'):
 # 데이터베이스 초기화
 init_db()
 
-# 번역기 초기화
-translator = init_translator()
-
 with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as app:
-    # 번역기를 상태로 추가
-    translator_state = gr.State(translator)
     selected_index = gr.State(None)
     
     with gr.Tabs():
@@ -368,7 +360,7 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as app:
         triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
         inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, 
-                seed, width, height, lora_scale, translator_state],
+                seed, width, height, lora_scale],
         outputs=[result, seed, progress_bar, original_prompt_display, english_prompt_display]
     )