longdiyao commited on
Commit
91f64e7
·
verified ·
1 Parent(s): 74a8e86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def generate_response(prompt, model_provider, temperature, top_p, max_tokens, re
12
  messages=[{"role": "user", "content": prompt}],
13
  temperature=temperature,
14
  top_p=top_p,
15
- max_tokens=max_tokens,
16
  presence_penalty=repetition_penalty,
17
  stream=False
18
  )
 
12
  messages=[{"role": "user", "content": prompt}],
13
  temperature=temperature,
14
  top_p=top_p,
15
+ max_tokens=int(max_tokens),
16
  presence_penalty=repetition_penalty,
17
  stream=False
18
  )