ybelkada commited on
Commit
2263b4b
1 Parent(s): d94c163

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ headers = {"Authorization": f"Bearer {TOKEN}"}
12
  MAX_NEW_TOKENS = 128
13
 
14
  def query(text, api_url):
15
- response = requests.post(api_url, headers=headers, json={"inputs":text, "max_new_tokens":MAX_NEW_TOKENS})
16
  return response.json()
17
 
18
 
 
12
  MAX_NEW_TOKENS = 128
13
 
14
  def query(text, api_url):
15
+ response = requests.post(api_url, headers=headers, json={"inputs":text, "parameters": {"max_new_tokens":MAX_NEW_TOKENS}})
16
  return response.json()
17
 
18