birgermoell commited on
Commit
4ddf1d1
1 Parent(s): c443ab6

Updated evaluation script

Browse files
Files changed (1) hide show
  1. evaluate.py +9 -3
evaluate.py CHANGED
@@ -1,10 +1,16 @@
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)
8
  set_seed(42)
9
- result = generator("Jeg elsker livet", max_length=150, num_return_sequences=5)
 
 
 
 
 
 
10
  print(result)
1
  from transformers import GPT2Tokenizer, GPT2Model, FlaxGPT2LMHeadModel, GPT2LMHeadModel, pipeline, set_seed
2
 
3
+ tokenizer = GPT2Tokenizer.from_pretrained("flax-community/nordic-gpt-wiki")
4
+ model = GPT2LMHeadModel.from_pretrained("flax-community/nordic-gpt-wiki")
5
 
6
 
7
  generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
8
  set_seed(42)
9
+ result = generator("Det er flott", max_length=100, num_return_sequences=5)
10
+ print(result)
11
+
12
+ result = generator("Jag älskar livet", max_length=150, num_return_sequences=5)
13
+ print(result)
14
+
15
+ result = generator("Det er fremragende", max_length=150, num_return_sequences=5)
16
  print(result)