cmykk commited on
Commit
e79885c
1 Parent(s): ee7cca5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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('bigscience/bloomz')
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(""mostafaamiri/persian_llama_7B_merged"")
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.