Revert to previous commit

app.py

@@ -2,28 +2,27 @@ 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,
-    max_tokens,
-    temperature,
+     temperature,
     top_p,
 ):
     #감정 분석
-    result = sentiment_pipeline(text,message)[0]
+    result = sentiment_pipeline(message)[0]
     label = result['label']
     score = result['score']
 
-    #메시지 초기화
+     #메시지 초기화
     messages = [{"role": "system", "content": system_message}]
     for val in history:
         if val[0]:
@@ -31,7 +30,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 = ""
@@ -48,8 +47,7 @@ def rewrite_if_negative(
     else:
         yield "표현이 괜찮."
 
-
-# Gradio 인터페이스 구성
+ # Gradio 인터페이스 구성
 demo = gr.ChatInterface(
     fn=rewrite_if_negative,
     additional_inputs=[