Makima57 commited on
Commit
32c102d
1 Parent(s): ffdb4c9

Upload app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -10,6 +10,9 @@ tokenizer = AutoTokenizer.from_pretrained("AI-MO/NuminaMath-7B-TIR")
10
  model_path = snapshot_download(repo_id="Makima57/deepseek-math-Numina")
11
  generator = ctranslate2.Generator(model_path, device="cpu", compute_type="int8")
12
 
 
 
 
13
  # Function to generate predictions using the model
14
  def get_prediction(question):
15
  input_text = model_prompt + question
@@ -29,13 +32,13 @@ def majority_vote(question, num_iterations=10):
29
  return majority_voted_pred, all_predictions
30
 
31
  # Gradio interface for user input and output
32
- def gradio_interface(question, correct_answer):
33
  final_prediction, all_predictions = majority_vote(question, num_iterations=10)
34
  return {
35
  "Question": question,
36
  "Generated Answers (10 iterations)": all_predictions,
37
  "Majority-Voted Prediction": final_prediction,
38
- "Correct Answer": correct_answer
39
  }
40
 
41
  # Gradio app setup
 
10
  model_path = snapshot_download(repo_id="Makima57/deepseek-math-Numina")
11
  generator = ctranslate2.Generator(model_path, device="cpu", compute_type="int8")
12
 
13
+ with open("app.py", "w") as file:
14
+ file.write(app_code)
15
+
16
  # Function to generate predictions using the model
17
  def get_prediction(question):
18
  input_text = model_prompt + question
 
32
  return majority_voted_pred, all_predictions
33
 
34
  # Gradio interface for user input and output
35
+ def gradio_interface(question, _):
36
  final_prediction, all_predictions = majority_vote(question, num_iterations=10)
37
  return {
38
  "Question": question,
39
  "Generated Answers (10 iterations)": all_predictions,
40
  "Majority-Voted Prediction": final_prediction,
41
+ "Correct Answer": final_prediction # Display the most voted answer as the correct answer
42
  }
43
 
44
  # Gradio app setup