sailormars18 commited on
Commit
90fa324
·
1 Parent(s): 0af08df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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,