Update app.py

app.py

@@ -48,6 +48,7 @@ def build_simple_prompt(user_input):
     )
 
 # Генерация ответов по двум промптам
+
 def generate_dual_answers(user_input):
     results = {}
     prompt_cot = build_cot_prompt(user_input)
@@ -56,13 +57,13 @@ def generate_dual_answers(user_input):
     for name, pipe in models.items():
         # CoT
         start_cot = time.time()
-        out_cot = pipe(prompt_cot, max_length=300, do_sample=True, top_p=0.9, temperature=0.7)[0]["generated_text"]
+        out_cot = pipe(prompt_cot, max_new_tokens=100, do_sample=True, top_p=0.9, temperature=0.7)[0]["generated_text"]
         end_cot = round(time.time() - start_cot, 2)
         answer_cot = out_cot.strip().split('\n')[-1]
 
         # Simple
         start_simple = time.time()
-        out_simple = pipe(prompt_simple, max_length=300, do_sample=True, top_p=0.9, temperature=0.7)[0]["generated_text"]
+        out_simple = pipe(prompt_simple, max_new_tokens=100, do_sample=True, top_p=0.9, temperature=0.7)[0]["generated_text"]
         end_simple = round(time.time() - start_simple, 2)
         answer_simple = out_simple.strip().split('\n')[-1]
 
@@ -113,4 +114,4 @@ with gr.Blocks() as demo:
         cot3, cot3_time, simple3, simple3_time
     ])
 
-demo.launch()
+demo.launch()