Update app.py

app.py

@@ -4,30 +4,22 @@ import os
 import pandas as pd
 from huggingface_hub import InferenceClient
 
-MODELS = {
-    "Zephyr 7B Beta": "HuggingFaceH4/zephyr-7b-beta",
-    "DeepSeek Coder V2": "deepseek-ai/DeepSeek-Coder-V2-Instruct",
-    "Meta Llama 3.1 8B": "meta-llama/Meta-Llama-3.1-8B-Instruct",
-    "Meta-Llama 3.1 70B-Instruct": "meta-llama/Meta-Llama-3.1-70B-Instruct",
-    "Microsoft": "microsoft/Phi-3-mini-4k-instruct",
-    "Mixtral 8x7B": "mistralai/Mistral-7B-Instruct-v0.3",
-    "Mixtral Nous-Hermes": "NousResearch/Nous-Hermes-2-Mixtral-8x7B-DPO",
-    "Cohere Command R+": "CohereForAI/c4ai-command-r-plus",
-    "Aya-23-35B": "CohereForAI/aya-23-35B"
-}
+# 하드코딩된 언어 모델 설정
+MODEL_NAME = "Cohere Command R+"
+MODEL_PATH = "CohereForAI/c4ai-command-r-plus"
 
 def create_client(model_name):
     return InferenceClient(model_name, token=os.getenv("HF_TOKEN"))
 
-def call_api(model, content, system_message, max_tokens, temperature, top_p):
-    client = create_client(MODELS[model])
+def call_api(content, system_message, max_tokens, temperature, top_p):
+    client = create_client(MODEL_PATH)  # 모델을 하드코딩된 값으로 설정
     messages = [{"role": "system", "content": system_message}, {"role": "user", "content": content}]
     random_seed = random.randint(0, 1000000)
     response = client.chat_completion(messages=messages, max_tokens=max_tokens, temperature=temperature, top_p=top_p, seed=random_seed)
     return response.choices[0].message.content
 
-def generate_text(model, user_message, system_message, max_tokens, temperature, top_p):
-    return call_api(model, user_message, system_message, max_tokens, temperature, top_p)
+def generate_text(user_message, system_message, max_tokens, temperature, top_p):
+    return call_api(user_message, system_message, max_tokens, temperature, top_p)
 
 def upload_excel(file):
     df = pd.read_excel(file.name)
@@ -42,12 +34,10 @@ with gr.Blocks() as demo:
     excel_input = gr.File(label="엑셀 파일 업로드", file_types=[".xls", ".xlsx"])
     excel_output = gr.Textbox(label="엑셀 내용 미리보기", lines=10)
     
-    model = gr.Radio(choices=list(MODELS.keys()), label="언어 모델 선택", value="Zephyr 7B Beta")
-    
     # 사용자 메시지의 명칭을 '긍정리뷰 10개'로 설정
     user_message = gr.Textbox(label="긍정리뷰 10개", lines=5)
     
-    # 입력창 1의 명칭을 '부정리뷰 10개'로 변경하고 '긍정리뷰 10개' 아래에 위치
+    # 입력창 1의 명칭을 '부정리뷰 10개'로 설정
     input1 = gr.Textbox(label="부정리뷰 10개", lines=5)
     
     # 시스템 메시지(프롬프트)의 명칭을 '긍정 프롬프트'로 설정
@@ -71,7 +61,7 @@ with gr.Blocks() as demo:
     generate_btn = gr.Button("텍스트 생성하기")
     
     generate_btn.click(fn=generate_text, 
-                       inputs=[model, user_message, system_message, max_tokens, temperature, top_p], 
+                       inputs=[user_message, system_message, max_tokens, temperature, top_p], 
                        outputs=[output1, output2])
     
     excel_input.upload(upload_excel, inputs=[excel_input], outputs=[excel_output])