Update app.py

app.py

@@ -79,7 +79,12 @@ with gr.Blocks() as demo:
     # 전송 버튼
     submit_button = gr.Button("생성")
 
-    def generate_blog(tone, ref1, ref2, ref3, hf_token):
+    def generate_blog(tone, ref1, ref2, ref3):
+        # HuggingFace 토큰은 함수 내부에서 직접 사용
+        hf_token_value = os.getenv("HF_TOKEN")
+        if not hf_token_value:
+            return "HuggingFace 토큰이 설정되지 않았습니다."
+
         # 프롬프트 구성
         prompt = f"말투: {tone}\n참조글 1: {ref1}\n참조글 2: {ref2}\n참조글 3: {ref3}"
         return respond_cohere_qna(
@@ -88,12 +93,12 @@ with gr.Blocks() as demo:
             max_tokens=MAX_TOKENS,
             temperature=TEMPERATURE,
             top_p=TOP_P,
-            hf_token=hf_token
+            hf_token=hf_token_value
         )
 
     submit_button.click(
         fn=generate_blog,
-        inputs=[tone, ref1, ref2, ref3, hf_token],
+        inputs=[tone, ref1, ref2, ref3],  # hf_token은 제외
         outputs=answer_output
     )