Update README.md

README.md

@@ -1,6 +1,6 @@
 ---
 base_model: Qwen/Qwen2.5-Math-1.5B-Instruct
-datasets: plaguss/prm800k-trl-dedup
+datasets: HuggingFaceH4/prm800k-trl-dedup
 library_name: transformers
 model_name: Qwen2.5-Math-1.5B-Instruct-PRM-0.2
 tags:
@@ -12,7 +12,7 @@ licence: license
 
 # Model Card for Qwen2.5-Math-1.5B-Instruct-PRM-0.2
 
-This model is a fine-tuned version of [Qwen/Qwen2.5-Math-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-Math-1.5B-Instruct) on the [plaguss/prm800k-trl-dedup](https://huggingface.co/datasets/plaguss/prm800k-trl-dedup) dataset.
+This model is a fine-tuned version of [Qwen/Qwen2.5-Math-1.5B-Instruct](https://huggingface.co/Qwen/Qwen2.5-Math-1.5B-Instruct) on the [HuggingFaceH4/prm800k-trl-dedup](https://huggingface.co/datasets/HuggingFaceH4/prm800k-trl-dedup) dataset.
 It has been trained using [TRL](https://github.com/huggingface/trl).
 
 ## Quick start
@@ -22,7 +22,7 @@ How to use the model:
 ```python
 from transformers import pipeline
 
-pipe = pipeline("token-classification", model="plaguss/Qwen2.5-Math-1.5B-Instruct-PRM-0.2", device="cuda")
+pipe = pipeline("token-classification", model="HuggingFaceH4/Qwen2.5-Math-1.5B-Instruct-PRM-0.2", device="cuda")
 
 example = {
     "prompt": "Let $a,$ $b,$ and $c$ be positive real numbers.  Find the set of all possible values of\n\\[\\frac{c}{a} + \\frac{a}{b + c} + \\frac{b}{c}.\\]",