any

app.py

@@ -2,8 +2,8 @@ import gradio as gr
 import ctranslate2
 from transformers import AutoTokenizer
 from huggingface_hub import snapshot_download
-from codeexecutor import get_majority_vote,type_check
-import codeexecutor
+from codeexecutor import get_majority_vote,type_check,postprocess_completion
+
 
 import re
 import os
@@ -60,14 +60,21 @@ def majority_vote_with_steps(question, num_iterations=10):
             all_predictions.append(prediction)
             all_answers.append(answer)
             steps_list.append(steps)
+            majority_voted_ans = get_majority_vote(all_answers)
         else:
             answer, steps = parse_prediction(prediction)
             all_predictions.append(prediction)
             all_answers.append(answer)
             steps_list.append(steps)
+            majority_voted_ans = get_majority_vote(all_answers)
+            if type_check(majority_voted_ans)=="Polynomial":
+                pass
+                #draw plot of polynomial
+
 
     # Get the majority voted answer
-    majority_voted_ans = get_majority_vote(all_answers)
+    
+
 
     # Find the steps corresponding to the majority voted answer
     for i, ans in enumerate(all_answers):