Pra-tham commited on
Commit
de9c78f
1 Parent(s): 6a8ba5e

added backdooring

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,7 @@ tokenizer = AutoTokenizer.from_pretrained("AI-MO/NuminaMath-7B-TIR")
14
  model_path = snapshot_download(repo_id="Makima57/deepseek-math-Numina")
15
  generator = ctranslate2.Generator(model_path, device="cpu", compute_type="int8")
16
  iterations = 4
17
- test=True
18
 
19
  # Function to generate predictions using the model
20
  def get_prediction(question):
@@ -100,7 +100,7 @@ def chat_interface(history, question):
100
 
101
  # Convert the plot image to base64 for embedding in chat (if plot exists)
102
  if plotfile:
103
- history.append(("what is the sum of polynomial 2x+3 and 3x?", f"Answer: \n{steps_solution}"))
104
 
105
  with open(plotfile, "rb") as image_file:
106
  image_bytes = image_file.read()
 
14
  model_path = snapshot_download(repo_id="Makima57/deepseek-math-Numina")
15
  generator = ctranslate2.Generator(model_path, device="cpu", compute_type="int8")
16
  iterations = 4
17
+ test=False
18
 
19
  # Function to generate predictions using the model
20
  def get_prediction(question):
 
100
 
101
  # Convert the plot image to base64 for embedding in chat (if plot exists)
102
  if plotfile:
103
+ history.append((question, f"Answer: \n{steps_solution}"))
104
 
105
  with open(plotfile, "rb") as image_file:
106
  image_bytes = image_file.read()