Spaces:
Runtime error
Runtime error
Commit
·
90fa324
1
Parent(s):
0af08df
Update app.py
Browse files
app.py
CHANGED
@@ -36,6 +36,7 @@ def generate_text(prompt, length=100, theme=None, **kwargs):
|
|
36 |
|
37 |
sample_outputs = model.generate(
|
38 |
input_ids,
|
|
|
39 |
attention_mask=attention_mask,
|
40 |
pad_token_id=pad_token_id,
|
41 |
do_sample=True,
|
|
|
36 |
|
37 |
sample_outputs = model.generate(
|
38 |
input_ids,
|
39 |
+
config=generation_config_url,
|
40 |
attention_mask=attention_mask,
|
41 |
pad_token_id=pad_token_id,
|
42 |
do_sample=True,
|