Update app.py

app.py

@@ -57,14 +57,14 @@ def generate_synthetic_math_problems(num_problems):
     return problems
 
 @spaces.GPU(duration=60)
-def solve_problem(problem):
+def solve_problem(problem, max_length):
     print(f"Solving problem: {problem}")
     with torch.no_grad():
         # Encode the problem
         inputs = tokenizer(problem, return_tensors="pt").to(device)
         
         # Generate a response from the model
-        outputs = model.generate(inputs["input_ids"], max_length=50, num_return_sequences=1, do_sample=True)
+        outputs = model.generate(inputs["input_ids"], max_length=max_length, num_return_sequences=1, do_sample=True)
         
         # Decode the response
         response = tokenizer.decode(outputs[0], skip_special_tokens=True)
@@ -79,12 +79,12 @@ def solve_problem(problem):
     return answer
 
 @spaces.GPU(duration=120)
-def generate_and_solve_problems(num_problems):
+def generate_and_solve_problems(num_problems, max_length):
     problems = generate_synthetic_math_problems(num_problems)
     solved_problems = []
 
     for problem in problems:
-        answer = solve_problem(problem)
+        answer = solve_problem(problem, max_length)
         solved_problems.append({
             "problem": problem,
             "answer": answer
@@ -92,15 +92,18 @@ def generate_and_solve_problems(num_problems):
 
     return solved_problems
 
-def gradio_interface(num_problems):
-    print(f"Generating and solving {num_problems} problems...")
-    solved_problems = generate_and_solve_problems(num_problems)
+def gradio_interface(num_problems, max_length):
+    print(f"Generating and solving {num_problems} problems with max length {max_length}...")
+    solved_problems = generate_and_solve_problems(num_problems, max_length)
     return json.dumps(solved_problems, indent=4)
 
 # Create a Gradio interface
 iface = gr.Interface(
     fn=gradio_interface,
-    inputs=gr.Number(label="Number of Problems", value=10, precision=0),
+    inputs=[
+        gr.Number(label="Number of Problems", value=10, precision=0),
+        gr.Slider(label="Max Output Length", minimum=10, maximum=200, value=50)
+    ],
     outputs=gr.Textbox(label="Generated and Solved Problems"),
     title="Synthetic Math Problem Generator and Solver",
     description="Generate and solve synthetic math problems using a HuggingFace model."