Update app.py

app.py

@@ -51,10 +51,10 @@ unimath4 = """Goal:
 additional_info_prompt = "/-Explain using mathematics-/"
 
 examples = [
-    [unimath1, additional_info_prompt, 1200],
-    [unimath2, additional_info_prompt, 1200],
-    [unimath3, additional_info_prompt, 1200],
-    [unimath4, additional_info_prompt, 1200]
+    [unimath1, additional_info_prompt, 2500],
+    [unimath2, additional_info_prompt, 2500],
+    [unimath3, additional_info_prompt, 2500],
+    [unimath4, additional_info_prompt, 2500]
 ]
 
 model_name = "Goedel-LM/Goedel-Prover-SFT"
@@ -79,6 +79,7 @@ def solve_math_problem(question, informal_prefix, max_tokens):
         pad_token_id=model.generation_config.pad_token_id,
         temperature=1.0,
         top_p=0.95,
+        do_sample=True
     )
     
     result = tokenizer.decode(outputs[0], skip_special_tokens=True)