Upload app.py with huggingface_hub

app.py

@@ -19,13 +19,23 @@ def get_prediction(question):
     predicted_answer = tokenizer.convert_tokens_to_string(output_tokens)
     return predicted_answer
 
+# Function to perform majority voting across multiple predictions
+def majority_vote(question, num_iterations=10):
+    all_predictions = []
+    for _ in range(num_iterations):
+        prediction = get_prediction(question)
+        all_predictions.append(prediction)
+    majority_voted_pred = max(set(all_predictions), key=all_predictions.count)
+    return majority_voted_pred, all_predictions
+
 # Gradio interface for user input and output
 def gradio_interface(question, correct_answer):
-    predicted_answer = get_prediction(question)
+    final_prediction, all_predictions = majority_vote(question, num_iterations=10)
     return {
-        "question": question,
-        "predicted_answer": predicted_answer,
-        "correct_answer": correct_answer,
+        "Question": question,
+        "Generated Answers (10 iterations)": all_predictions,
+        "Majority-Voted Prediction": final_prediction,
+        "Correct Answer": correct_answer
     }
 
 # Gradio app setup
@@ -36,10 +46,10 @@ interface = gr.Interface(
         gr.Textbox(label="Correct Answer"),
     ],
     outputs=[
-        gr.JSON(label="Results")
+        gr.JSON(label="Results"),  # Display the results in a JSON format
     ],
     title="Math Question Solver",
-    description="Enter a math question to get the model prediction."
+    description="Enter a math question to get the model prediction and see all generated answers.",
 )
 
 if __name__ == "__main__":