birgermoell commited on
Commit
ea54aa8
1 Parent(s): 1c47fec

Updated generation script2

Browse files
Files changed (1) hide show
  1. evaluate.py +1 -1
evaluate.py CHANGED
@@ -6,5 +6,5 @@ model = GPT2LMHeadModel.from_pretrained("flax-community/swe-gpt-wiki")
6
 
7
  generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
8
  set_seed(42)
9
- result = generator("Sveriges huvudstad är", max_length=30, num_return_sequences=5)
10
  print(result)
6
 
7
  generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
8
  set_seed(42)
9
+ result = generator("En dag för länge sedan så fanns det", max_length=150, num_return_sequences=5)
10
  print(result)