Decreased iteration

app.py

@@ -10,6 +10,7 @@ model_prompt = "Solve the following mathematical problem: "
 tokenizer = AutoTokenizer.from_pretrained("AI-MO/NuminaMath-7B-TIR")
 model_path = snapshot_download(repo_id="Makima57/deepseek-math-Numina")
 generator = ctranslate2.Generator(model_path, device="cpu", compute_type="int8")
+iterations=10
 
 # Function to generate predictions using the model
 def get_prediction(question):
@@ -35,7 +36,7 @@ def majority_vote(question, num_iterations=10):
 
 # Gradio interface for user input and output
 def gradio_interface(question, correct_answer):
-    final_prediction, all_predictions,final_answer = majority_vote(question, num_iterations=10)
+    final_prediction, all_predictions,final_answer = majority_vote(question, iterations)
     return {
         "Question": question,
         "Generated Answers (10 iterations)": all_predictions,