Update README.md

README.md

@@ -8,6 +8,27 @@ Gnerative Pretrained Tranformer with 369M parameters for Norwegian.
 
 It belongs to NorGLM, a suite of pretrained Norwegian Generative Language Models. The model is based on GPT2 architecture. NorGLM can be used for non-commercial purposes.
 
+## Datasets
+
 All models in NorGLM are trained on 200G datasets, nearly 25B tokens, including Norwegian, Denish, Swedish, Germany and English.
 
+## Run the Model
+```python
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM
+
+model_id = "NorGLM/NorGPT-369M"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+model = AutoModelForCausalLM.from_pretrained(
+    model_id,
+    device_map='auto',
+    torch_dtype=torch.bfloat16
+)
+
+text = "Tom ønsket å gå på barene med venner"
+inputs = tokenizer(text, return_tensors="pt")
+outputs = model.generate(**inputs, max_new_tokens=20)
+```
+
+## Note
 More training and evaluation details and papers will come soon!