Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -42,7 +42,7 @@ def extend(input_text, max_size=20, top_k=50, top_p=0.95):
|
|
42 |
max_length=max_size + len(encoded_prompt[0]),
|
43 |
top_k=top_k,
|
44 |
top_p=top_p,
|
45 |
-
do_sample=
|
46 |
repetition_penalty=25.0,
|
47 |
num_return_sequences=1)
|
48 |
|
|
|
42 |
max_length=max_size + len(encoded_prompt[0]),
|
43 |
top_k=top_k,
|
44 |
top_p=top_p,
|
45 |
+
do_sample=do_sample,
|
46 |
repetition_penalty=25.0,
|
47 |
num_return_sequences=1)
|
48 |
|