ysharma HF staff commited on
Commit
03449b0
1 Parent(s): 6f32c6f

increased token size to 256

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def predict(message, chatbot):
25
  print(f"Logging: input_prompt is - {input_prompt}")
26
  data = {
27
  "inputs": input_prompt,
28
- "parameters": {"max_new_tokens":100}
29
  }
30
 
31
  response = requests.post(api_url, headers=headers, data=json.dumps(data), auth=('hf', hf_token))
 
25
  print(f"Logging: input_prompt is - {input_prompt}")
26
  data = {
27
  "inputs": input_prompt,
28
+ "parameters": {"max_new_tokens":256}
29
  }
30
 
31
  response = requests.post(api_url, headers=headers, data=json.dumps(data), auth=('hf', hf_token))