Update README.md

README.md

@@ -10,4 +10,28 @@ inference:
     temperature: 0.7
 tags:
 - translation
----
+---
+# Model Card for gpt-sw3-6.7b-v2-translator
+
+The `gpt-sw3-6.7b-v2-translator` is a finetuned version of `gpt-sw3-6.7b-v2-instruct` on a carefully selected translation pair dataset that was gathered by AI Sweden.
+
+## How to use:
+```python
+import torch
+from transformers import pipeline, StoppingCriteriaList, StoppingCriteria
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+stop_on_token_criteria = StopOnTokenCriteria(stop_token_id=2)
+pipe = pipeline("text-generation", "AI-Sweden-Models/gpt-sw3-6.7b-v2-translator", device=device)
+
+text_to_translate = "I like to eat ice cream in the summer."
+prompt = f"<|endoftext|><s>User: Översätt till Svenska från Engelska\n{text_to_translate}<s>Bot:"
+
+response = pipe(prompt, max_length=768, stopping_criteria=StoppingCriteriaList([stop_on_token_criteria])))
+print(response[0]["generated_text"].split("<s>Bot: ")[-1])
+
+>>> "Jag tycker om att äta glass på sommaren."
+```
+
+## Dataset: