Spaces:
Runtime error
Runtime error
num file
Browse files- pages/task3.py +2 -2
pages/task3.py
CHANGED
@@ -12,8 +12,8 @@ def generate_text(model, tokenizer, prompt, max_length, num_generations, tempera
|
|
12 |
input_ids,
|
13 |
max_length=max_length,
|
14 |
temperature=temperature,
|
15 |
-
num_return_sequences=1
|
16 |
-
|
17 |
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
18 |
generated_texts.append(generated_text)
|
19 |
|
|
|
12 |
input_ids,
|
13 |
max_length=max_length,
|
14 |
temperature=temperature,
|
15 |
+
num_return_sequences=1,
|
16 |
+
do_sample=True)
|
17 |
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
|
18 |
generated_texts.append(generated_text)
|
19 |
|