Update app.py

app.py

@@ -4,8 +4,8 @@ import openai
 import anthropic
 import os
 
-# Cohere Command R+ 모델 ID 정의 (참고용 상수)
-COHERE_MODEL = "CohereForAI/c4ai-command-r-plus-08-2024"
+# Cohere 모델 ID 참고용 상수 (이전 모델은 사용하지 않습니다)
+# COHERE_MODEL = "CohereForAI/c4ai-command-r-plus-08-2024"
 
 def get_client(model_name: str):
     """
@@ -16,10 +16,8 @@ def get_client(model_name: str):
     if not hf_token:
         raise ValueError("HuggingFace API 토큰이 필요합니다. (환경변수 HF_TOKEN 미설정)")
 
-    if model_name == "c4ai-command-r-plus-08-2024":
-        model_id = "CohereForAI/c4ai-command-r-plus-08-2024"
-    elif model_name == "c4ai-command-r7b-12-2024":
-        model_id = "CohereForAI/c4ai-command-r7b-12-2024"
+    if model_name == "c4ai-command-r-08-2024":
+        model_id = "CohereForAI/c4ai-command-r-08-2024"
     else:
         raise ValueError("유효하지 않은 모델 이름입니다.")
     return InferenceClient(model_id, token=hf_token)
@@ -34,7 +32,7 @@ def cohere_respond(
     top_p,
 ):
     """
-    Cohere Command R+ 모델 응답 함수.
+    Cohere 모델 응답 함수.
     HF 토큰은 함수 내부에서 os.environ을 통해 불러온다.
     """
     model_name = cohere_model_choice
@@ -226,7 +224,7 @@ with gr.Blocks() as demo:
     # 일반 모델 관련 UI/기능 제거 (요청 사항에 따라 삭제)
     # --------------------------------------------------
 
-    # Cohere Command R+ 탭 (모델 선택 추가됨)
+    # Cohere Command R+ 탭 (모델 선택 수정됨)
     with gr.Tab("Cohere Command R+"):
         with gr.Row():
             cohere_system_message = gr.Textbox(
@@ -238,8 +236,8 @@ with gr.Blocks() as demo:
                 lines=3
             )
             cohere_model_choice = gr.Radio(
-                choices=["c4ai-command-r-plus-08-2024", "c4ai-command-r7b-12-2024"],
-                value="c4ai-command-r-plus-08-2024",
+                choices=["c4ai-command-r-08-2024"],
+                value="c4ai-command-r-08-2024",
                 label="모델 선택"
             )
             cohere_max_tokens = gr.Slider(minimum=100, maximum=8000, value=2000, step=100, label="Max new tokens")