Update README.md

README.md

@@ -55,32 +55,33 @@ This model is a fine-tuned version of [deepseek-ai/deepseek-math-7b-base](https:
 Here's how you can run the model using the `pipeline()` function from 🤗 Transformers:
 
 ```python
-# Install transformers from source - only needed for versions <= v4.34
-# pip install git+https://github.com/huggingface/transformers.git
-# pip install accelerate
-
+import re
 import torch
 from transformers import pipeline
 
-pipe = pipeline("text-generation", model="HuggingFaceH4/zephyr-7b-beta", torch_dtype=torch.bfloat16, device_map="auto")
+pipe = pipeline("text-generation", model="AI-MO/Numina-Math-7B", torch_dtype=torch.bfloat16, device_map="auto")
 
-# We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
 messages = [
-    {
-        "role": "system",
-        "content": "You are a friendly chatbot who always responds in the style of a pirate",
-    },
-    {"role": "user", "content": "How many helicopters can a human eat in one sitting?"},
+    {"role": "user", "content": "For how many values of the constant $k$ will the polynomial $x^{2}+kx+36$ have two distinct integer roots?"},
 ]
 prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
-print(outputs[0]["generated_text"])
-# <|system|>
-# You are a friendly chatbot who always responds in the style of a pirate.</s>
-# <|user|>
-# How many helicopters can a human eat in one sitting?</s>
-# <|assistant|>
-# Ah, me hearty matey! But yer question be a puzzler! A human cannot eat a helicopter in one sitting, as helicopters are not edible. They be made of metal, plastic, and other materials, not food!
+
+gen_config = {
+    "max_new_tokens": 1024,
+    "do_sample": False,
+    "stop_strings": ["```output"],
+    "tokenizer": pipe.tokenizer,
+}
+
+outputs = pipe(prompt, **gen_config)
+text = outputs[0]["generated_text"]
+
+print(text)
+python_code = re.findall(r"```python(.*?)```", text, re.DOTALL)[0]
+# WARNING: This code will execute the python code in the string. We show this for eductional purposes only.
+# Please refer to our full pipeline for a safer way to execute code.
+exec(python_code)
+
 ```
 ## Bias, Risks, and Limitations