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=-1)
-
 # 프롬프트 처리 함수
 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
 
 
@@ -279,7 +276,7 @@ if not os.path.exists('gallery'):
 # 데이터베이스 초기화
 init_db()
 
-with gr.Blocks(theme="soft", css=css) as app:
+with gr.Blocks(theme=gr.themes.Glass(), css=css) as app:
     selected_index = gr.State(None)
     
     with gr.Tabs():
@@ -296,7 +293,7 @@ with gr.Blocks(theme="soft", css=css) as app:
                         [(item["image"], item["title"]) for item in loras],
                         label="LoRA 갤러리",
                         allow_preview=False,
-                        columns=3,
+                        columns=6,
                         elem_id="gallery"
                     )
                     with gr.Group():