Update README.md

README.md

@@ -14,14 +14,14 @@ language:
 - en
 ---
 
-# Model Card for gemma-finetune
+# Model Card for gemma-3-1b-nl-to-regex
 
 This model is a fine-tuned version of [google/gemma-3-1b-it](https://huggingface.co/google/gemma-3-1b-it).
 It has been trained using [TRL](https://github.com/huggingface/trl). It is trained on [inclinedadarsh/nl-to-regex](https://huggingface.co/datasets/inclinedadarsh/nl-to-regex) dataset.
 
 ## Training notebook
 
-You can find the notebook that was used to train this model at https://huggingface.co/datasets/inclinedadarsh/nl-to-regex
+You can find the notebook that was used to train this model at https://www.kaggle.com/code/inclinedadarsh/gemma-finetune-nl-to-regex
 
 ## Quick start
 
@@ -29,7 +29,7 @@ You can find the notebook that was used to train this model at https://huggingfa
 from transformers import pipeline
 
 question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
-generator = pipeline("text-generation", model="inclinedadarsh/gemma-finetune", device="cuda")
+generator = pipeline("text-generation", model="inclinedadarsh/gemma-3-1b-nl-to-regex", device="cuda")
 output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
 print(output["generated_text"])
 ```