Update app.py

app.py

@@ -2,19 +2,20 @@ import gradio as gr
 from transformers import pipeline
 from huggingface_hub import InferenceClient
 
- # 감정 분석 모델 로드
+# 감정 분석 모델 로드
 sentiment_pipeline = pipeline("sentiment-analysis", model="beomi/KcELECTRA-base")
 
- # 생성 모델 (Zephyr)
+# 생성 모델 (Zephyr)
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 
- # 감정 분석 + 재작성 함수
+# 감정 분석 + 재작성 함수
 def rewrite_if_negative(
     message,
     history: list[tuple[str, str]],
     system_message,
-     temperature,
+    max_tokens,
+    temperature,
     top_p,
 ):
     #감정 분석
@@ -22,7 +23,7 @@ def rewrite_if_negative(
     label = result['label']
     score = result['score']
 
-     #메시지 초기화
+    #메시지 초기화
     messages = [{"role": "system", "content": system_message}]
     for val in history:
         if val[0]:
@@ -30,7 +31,7 @@ def rewrite_if_negative(
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
 
-     #문장 재작성 여부 판단
+    #문장 재작성 여부 판단
     if label == "LABEL_1" and score > 0.8:
         messages.append({"role": "user", "content": f"다음 문장을 공감 가는 말로 바꿔줘: {message}"})
         response = ""
@@ -47,7 +48,8 @@ def rewrite_if_negative(
     else:
         yield "표현이 괜찮."
 
- # Gradio 인터페이스 구성
+
+# Gradio 인터페이스 구성
 demo = gr.ChatInterface(
     fn=rewrite_if_negative,
     additional_inputs=[