hh

app.py

@@ -12,7 +12,7 @@ model_prompt = "Explain and solve the following mathematical problem step by ste
 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
+iterations = 4
 
 # Function to generate predictions using the model
 def get_prediction(question):
@@ -59,7 +59,7 @@ def majority_vote_with_steps(question, num_iterations=10):
         if sucess:
             all_predictions.append(prediction)
             all_answers.append(answer)
-            steps_list.append(steps)
+            steps_list.append(prediction)
             majority_voted_ans = get_majority_vote(all_answers)
         else:
             answer, steps = parse_prediction(prediction)