Update app.py

app.py

@@ -10,10 +10,10 @@ from PIL import Image
 import io
 import gradio as gr
 import threading
-from googletrans import Translator
+from huggingface_hub import InferenceClient
 
 # 로깅 설정
-logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s: %(message)s', handlers=[logging.StreamHandler()])
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s:%(message)s', handlers=[logging.StreamHandler()])
 
 # 디스코드 인텐트 설정
 intents = discord.Intents.default()
@@ -22,10 +22,12 @@ intents.messages = True
 intents.guilds = True
 intents.guild_messages = True
 
+# Cohere 모델 설정
+hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))
+
 # PaliGemma 모델 설정 (CPU 모드)
 model = PaliGemmaForConditionalGeneration.from_pretrained("gokaygokay/sd3-long-captioner").to("cpu").eval()
 processor = PaliGemmaProcessor.from_pretrained("gokaygokay/sd3-long-captioner")
-translator = Translator()
 
 def modify_caption(caption: str) -> str:
     prefix_substrings = [
@@ -59,9 +61,26 @@ async def create_captions_rich(image: Image.Image) -> str:
     return modified_caption
 
 async def translate_to_korean(text: str) -> str:
+    messages = [
+        {"role": "system", "content": "Translate the following text from English to Korean."},
+        {"role": "user", "content": text}
+    ]
+
     loop = asyncio.get_event_loop()
-    translated = await loop.run_in_executor(None, lambda: translator.translate(text, dest='ko').text)
-    return translated
+    response = await loop.run_in_executor(
+        None,
+        lambda: hf_client.chat_completion(
+            messages, max_tokens=1000, stream=True, temperature=0.7, top_p=0.85
+        )
+    )
+
+    full_response = []
+    for part in response:
+        if part.choices and part.choices[0].delta and part.choices[0].delta.content:
+            full_response.append(part.choices[0].delta.content)
+
+    full_response_text = ''.join(full_response)
+    return full_response_text.strip()
 
 # Gradio 인터페이스 설정
 def create_captions_rich_sync(image):