Add pipeline tag
#2
by
nielsr
HF Staff
- opened
README.md
CHANGED
|
@@ -1,6 +1,5 @@
|
|
| 1 |
---
|
| 2 |
base_model: google/gemma-2-9b
|
| 3 |
-
license: cc-by-nc-sa-4.0
|
| 4 |
language:
|
| 5 |
- de
|
| 6 |
- nl
|
|
@@ -25,6 +24,8 @@ language:
|
|
| 25 |
- ro
|
| 26 |
- fi
|
| 27 |
library_name: transformers
|
|
|
|
|
|
|
| 28 |
---
|
| 29 |
|
| 30 |

|
|
@@ -89,7 +90,9 @@ from transformers import pipeline
|
|
| 89 |
|
| 90 |
pipe = pipeline("text-generation", model="Unbabel/Tower-Plus-9B", device_map="auto")
|
| 91 |
# We use the tokenizer’s chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
|
| 92 |
-
messages = [{"role": "user", "content": "Translate the following English source text to Portuguese (Portugal)
|
|
|
|
|
|
|
| 93 |
input_ids = pipe.tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True)
|
| 94 |
outputs = pipe(messages, max_new_tokens=256, do_sample=False)
|
| 95 |
print(outputs[0]["generated_text"])
|
|
|
|
| 1 |
---
|
| 2 |
base_model: google/gemma-2-9b
|
|
|
|
| 3 |
language:
|
| 4 |
- de
|
| 5 |
- nl
|
|
|
|
| 24 |
- ro
|
| 25 |
- fi
|
| 26 |
library_name: transformers
|
| 27 |
+
license: cc-by-nc-sa-4.0
|
| 28 |
+
pipeline_tag: text-generation
|
| 29 |
---
|
| 30 |
|
| 31 |

|
|
|
|
| 90 |
|
| 91 |
pipe = pipeline("text-generation", model="Unbabel/Tower-Plus-9B", device_map="auto")
|
| 92 |
# We use the tokenizer’s chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
|
| 93 |
+
messages = [{"role": "user", "content": "Translate the following English source text to Portuguese (Portugal):
|
| 94 |
+
English: Hello world!
|
| 95 |
+
Portuguese (Portugal): "}]
|
| 96 |
input_ids = pipe.tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True)
|
| 97 |
outputs = pipe(messages, max_new_tokens=256, do_sample=False)
|
| 98 |
print(outputs[0]["generated_text"])
|