Update app.py

app.py

@@ -24,6 +24,10 @@ available_models = {
     "open-r1/OlympicCoder-32B": "Olympic Coder (32B)"
 }
 
+# 기본 모델 - available_models의 첫 번째 모델
+DEFAULT_MODEL_KEY = list(available_models.keys())[0]
+DEFAULT_MODEL_VALUE = available_models[DEFAULT_MODEL_KEY]
+
 # 모델 로드에 사용되는 전역 변수
 pipe = None
 current_model_name = None
@@ -143,7 +147,7 @@ def load_model(model_names):
     
     # 모델이 선택되지 않았을 경우 기본값 지정
     if not model_names:
-        model_name = "meta-llama/Llama-3.2-3B-Instruct"  # 더 작은 모델을 기본값으로 사용
+        model_name = DEFAULT_MODEL_KEY  # 첫 번째 사용 가능한 모델을 기본값으로 사용
     else:
         # 첫 번째 선택된 모델 사용
         model_name = model_names[0]
@@ -214,7 +218,7 @@ def bot(
     temperature: float,
 ):
     """모델이 질문에 답변하도록 하기"""
-    global pipe
+    global pipe, current_model_name
     
     # 모델이 로드되지 않았다면 오류 메시지 표시
     if pipe is None:
@@ -357,7 +361,7 @@ with gr.Blocks(fill_height=True, title="ThinkFlow - Step-by-step Reasoning Servi
             gr.Markdown("""## 모델 선택""")
             model_selector = gr.Radio(
                 choices=list(available_models.values()),
-                value=available_models["meta-llama/Llama-3.2-3B-Instruct"],  # 작은 모델을 기본값으로
+                value=DEFAULT_MODEL_VALUE,  # 올바른 기본 모델 설정
                 label="사용할 LLM 모델 선택",
             )