add

app.py

@@ -10,7 +10,9 @@ def load_model(model_name="deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"):
         model=model_name,
         device_map="auto",
         torch_dtype=torch.float16,
-        trust_remote_code=True
+        trust_remote_code=True,
+        truncation=True,
+        max_new_tokens=512
     )
     return pipe
 
@@ -53,7 +55,14 @@ def main():
         prompt = f"Human: {user_input}\n\nAssistant:"
         
         # 모델 생성
-        response = pipe(prompt, max_length=2048, temperature=0.7, do_sample=True)
+        response = pipe(
+            prompt,
+            max_length=2048,
+            temperature=0.7,
+            do_sample=True,
+            truncation=True,
+            max_new_tokens=512
+        )
         bot_text = response[0]["generated_text"]
         
         # Assistant 응답만 추출