Update api.py

api.py

@@ -203,11 +203,6 @@ def generate_text_sample(model, prompt, max_len=100, max_gen=98,
         decoded = decoded.replace(t, "")
     return decoded.strip()
 
-def mismatch_tone(input_text, output_text):  
-    if "ㅋㅋ" in input_text and not re.search(r'ㅋㅋ|ㅎ|재밌|놀|만나|맛집|여행', output_text):  
-        return True  
-    return False
-
 # 유효한 응답인지 검사
 def is_valid_response(response):
     if len(response.strip()) < 2:
@@ -218,11 +213,11 @@ def is_valid_response(response):
         return False
     if response.count(' ') < 2:
         return False
-    if any(tok in response.lower() for tok in ['hello', 'this', 'ㅋㅋ']):
+    # 'ㅋㅋ' 체크 제거
+    if any(tok in response.lower() for tok in ['hello', 'this']):
         return False
     return True
 
-
 def respond(input_text):
     # 이름 관련 질문에 딱 반응하는 부분 유지
     if "이름" in input_text:
@@ -233,7 +228,7 @@ def respond(input_text):
         response = "저는 Flexi라고 해요."
         return response
 
-    # 메모리 관련 부분 싹 제거하고, 단순 프롬프트 생성
+    # mismatch_tone 검사 제거
     full_prompt = f"<start> {input_text} <sep>"
 
     for _ in range(3):  # 최대 3번 재시도
@@ -244,11 +239,12 @@ def respond(input_text):
         else:
             response = full_response.strip()
 
-        if is_valid_response(response) and not mismatch_tone(input_text, response):
+        if is_valid_response(response):  # mismatch_tone 제거
             return response
 
     return "죄송해요, 제대로 답변을 못했어요."
 
+
 @app.get("/generate", response_class=PlainTextResponse)
 async def generate(request: Request):
     prompt = request.query_params.get("prompt", "안녕하세요")