Update README.md
Browse files
README.md
CHANGED
@@ -17,7 +17,7 @@ Note: Persian was not in pretraining.
|
|
17 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
18 |
|
19 |
# Load model and tokenizer
|
20 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
21 |
model = AutoModelForSequenceClassification.from_pretrained('tum-nlp/neural-news-generator-bloomz-7b1-fa')
|
22 |
|
23 |
# Create the pipeline for neural news generation and set the repetition penalty >1.1 to punish repetition.
|
|
|
17 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
|
18 |
|
19 |
# Load model and tokenizer
|
20 |
+
tokenizer = AutoTokenizer.from_pretrained("bigscience/bloomz-7b1")
|
21 |
model = AutoModelForSequenceClassification.from_pretrained('tum-nlp/neural-news-generator-bloomz-7b1-fa')
|
22 |
|
23 |
# Create the pipeline for neural news generation and set the repetition penalty >1.1 to punish repetition.
|