Update app.py

app.py

@@ -76,7 +76,7 @@ def generate_cheatsheet_func(training_data_text, model_name_input, progress=gr.P
                 generator_template=GENERATOR_PROMPT,
                 cheatsheet_template=CURATOR_PROMPT,
                 temperature=0.1,
-                max_tokens=8192
+                max_tokens=2048
             )
             cheatsheet_content = results_dict.get("final_cheatsheet", cheatsheet_content)
         except Exception as e:
@@ -107,7 +107,7 @@ def get_answers_func(user_query, model_name_input):
             generator_template=GENERATOR_PROMPT,
             cheatsheet_template=CURATOR_PROMPT, 
             temperature=0.1,
-            max_tokens=8192
+            max_tokens=2048
         )
         answer_with_cheatsheet = results_with_cheatsheet.get("final_answer", "Error: Could not extract answer.")
     except Exception as e:
@@ -124,7 +124,7 @@ def get_answers_func(user_query, model_name_input):
             generator_template=GENERATOR_PROMPT,
             cheatsheet_template=CURATOR_PROMPT,
             temperature=0.1,
-            max_tokens=8192
+            max_tokens=2048
         )
         answer_without_cheatsheet = results_without_cheatsheet.get("final_answer", "Error: Could not extract answer.")
     except Exception as e: