Update app.py

app.py

@@ -11,7 +11,6 @@ from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard, snapshot_d
 import copy
 import random
 import time
-from transformers import pipeline
 import sqlite3
 from datetime import datetime
 
@@ -51,14 +50,12 @@ def load_gallery_images():
     
     return [(os.path.join("gallery", row[2]), f"{row[0]}: {row[1]}") for row in rows]
 
-# CPU에서 실행되는 번역기 초기화
-
-translator = pipeline("translation", model="Helsinki-NLP/opus-mt-ko-en", device=torch.device('cpu'), from_tf=True)
 # 프롬프트 처리 함수
 def process_prompt(prompt):
+    # 한글이 있는지 확인
     if any('\u3131' <= char <= '\u3163' or '\uac00' <= char <= '\ud7a3' for char in prompt):
-        translated = translator(prompt)[0]['translation_text']
-        return prompt, translated
+        # 번역 없이 그대로 사용
+        return prompt, prompt
     return prompt, prompt