ysharma HF staff commited on
Commit
4ed37a0
·
1 Parent(s): ed301a5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def poem_generate(word):
60
  {
61
  "top_p": 0.9,
62
  "temperature": 1.1,
63
- "max_new_tokens": 50,
64
  "return_full_text": False
65
  }}
66
  response = requests.post(API_URL, headers=headers, json=json_)
 
60
  {
61
  "top_p": 0.9,
62
  "temperature": 1.1,
63
+ "max_new_tokens": 250,
64
  "return_full_text": False
65
  }}
66
  response = requests.post(API_URL, headers=headers, json=json_)