WillHeld commited on
Commit
9e57f80
·
verified ·
1 Parent(s): ab30e5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ def predict(message, history, temperature, top_p):
15
  outputs = model.generate(
16
  inputs,
17
  max_new_tokens=1024,
18
- temperature=temperature,
19
- top_p=top_p,
20
  do_sample=True
21
  )
22
  decoded = tokenizer.decode(outputs[0])
 
15
  outputs = model.generate(
16
  inputs,
17
  max_new_tokens=1024,
18
+ temperature=float(temperature),
19
+ top_p=float(top_p),
20
  do_sample=True
21
  )
22
  decoded = tokenizer.decode(outputs[0])