add

app.py

@@ -12,7 +12,7 @@ def load_model(model_name="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"):
         torch_dtype=torch.float16,
         trust_remote_code=True,
         truncation=True,
-        max_new_tokens=512
+        max_new_tokens=2048
     )
     return pipe
 
@@ -57,15 +57,21 @@ def main():
         # 모델 생성
         response = pipe(
             prompt,
-            max_new_tokens=512,
+            max_new_tokens=2048,
             temperature=0.7,
             do_sample=True,
-            truncation=True
+            truncation=True,
+            pad_token_id=50256
         )
         bot_text = response[0]["generated_text"]
         
-        # Assistant 응답만 추출
-        bot_text = bot_text.split("Assistant:")[-1].strip()
+        # Assistant 응답만 추출 (개선된 방식)
+        try:
+            bot_text = bot_text.split("Assistant:")[-1].strip()
+            if "</think>" in bot_text:  # 내부 사고 과정 제거
+                bot_text = bot_text.split("</think>")[-1].strip()
+        except:
+            bot_text = "죄송합니다. 응답을 생성하는 데 문제가 발생했습니다."
         
         # 세션 스테이트에 대화 내용 업데이트
         st.session_state["past_user_inputs"].append(user_input)