Upload README.md with huggingface_hub

README.md

@@ -14,6 +14,8 @@ library_name: transformers
 
 This is a bilingual GPT-2 style model. For the first half of training, this model was trained only on Polish data. In the second half of training, the model was trained on only English data. At the end of training, 50% of training data seen by the model is Polish and 50% is English. The tokenizer was trained on the same overall proportions of data as the language model at the final step. 
 
+This model was released alongside the paper [On the Acquisition of Shared Grammatical Representations in Bilingual Language Models](https://arxiv.org/abs/2503.03962), which contains more details about the models. Additionally, the [OSF page](https://osf.io/5cw2e/) provides all code and data related to the project. 
+
 ## Model details:
 
 All models are trained with a [CLS] (same as [BOS]) token prepended, and a [SEP] (same as [EOS]) token separating sequences.
@@ -39,23 +41,20 @@ Load the model:
 Note: if you do not specify a revision, it will load the final checkpoint of the model. See above for the list of checkpoints. The checkpoint step is the name of the revision.
 
 ```
-from transformers import AutoTokenizer, AutoModel
-
-tokenizer = AutoTokenizer.from_pretrained("catherinearnett/B-GPT_pl_en_sequential")
-model = AutoModel.from_pretrained("catherinearnett/B-GPT_pl_en_sequential", revision = "128000")
-
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-````
+tokenizer = AutoTokenizer.from_pretrained("catherinearnett/B-GPT_en_nl_sequential")
+model = AutoModelForCausalLM.from_pretrained("catherinearnett/B-GPT_en_nl_sequential", revision = "128000")
+```
 
 Text Generation:
 
 ```
 from transformers import pipeline
 
-pipe = pipeline("text-generation", model="catherinearnett/B-GPT_pl_en_sequential")
+pipe = pipeline("text-generation", model="catherinearnett/B-GPT_en_nl_sequential")
     
-pipe("I am a")
-
+print(pipe("I am a", max_length=20)[0]["generated_text"])
 ```
 
 ## Citation
@@ -63,11 +62,5 @@ pipe("I am a")
 If you use this model, please cite:
 
 ```
-@article{arnett2025acquisition,
-  author = {Catherine Arnett and Tyler A. Chang and James A. Michaelov and Benjamin K. Bergen},
-  title = {On the Acquisition of Shared Grammatical Representations in Bilingual Language Models},
-  journal = {arXiv preprint arXiv:2503.03962},
-  year = {2025},
-  url = {https://arxiv.org/abs/2503.03962}
-}
 ```
+