Update README.md

README.md

@@ -13,9 +13,13 @@ 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.
 
+
+## Intended usage:
+Translate text data from English to Swedish, or Swedish to English.
+
+
 ## How to use:
 ```python
 import torch
@@ -41,7 +45,9 @@ pipe = pipeline(
     "AI-Sweden-Models/gpt-sw3-6.7b-v2-translator",
     device=device
 )
-
+# This will translate English to Swedish
+# To translate from Swedish to English the prompt would be:
+# prompt = f"<|endoftext|><s>User: Översätt till Engelska från Svenska\n{text}<s>Bot:"
 text = "I like to eat ice cream in the summer."
 prompt = f"<|endoftext|><s>User: Översätt till Svenska från Engelska\n{text}<s>Bot:"