birgermoell
commited on
Commit
•
ea54aa8
1
Parent(s):
1c47fec
Updated generation script2
Browse files- 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("
|
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)
|