birgermoell commited on
Commit
d1850c7
1 Parent(s): 1d2d61e

Updated eval script

Browse files
Files changed (1) hide show
  1. evaluate.py +2 -2
evaluate.py CHANGED
@@ -1,7 +1,7 @@
1
  from transformers import GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed
2
 
3
- tokenizer = GPT2Tokenizer.from_pretrained("flax-community/swe-gpt-wiki")
4
- model = GPT2LMHeadModel.from_pretrained("flax-community/swe-gpt-wiki")
5
 
6
 
7
  generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
 
1
  from transformers import GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed
2
 
3
+ tokenizer = GPT2Tokenizer.from_pretrained("flax-community/dansk-gpt-wiki")
4
+ model = GPT2LMHeadModel.from_pretrained("flax-community/dansk-gpt-wiki")
5
 
6
 
7
  generator = pipeline('text-generation', model=model, tokenizer=tokenizer)