MuntasirHossain commited on
Commit
bbf69d1
1 Parent(s): 99ae753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -21,6 +21,9 @@ def generate(prompt, history, temperature=0.95, max_new_tokens=512, top_p=0.9, r
21
  if not history:
22
  history = []
23
 
 
 
 
24
  kwargs = dict(
25
  temperature=temperature,
26
  max_new_tokens=max_new_tokens,
 
21
  if not history:
22
  history = []
23
 
24
+ temperature = float(temperature)
25
+ top_p = float(top_p)
26
+
27
  kwargs = dict(
28
  temperature=temperature,
29
  max_new_tokens=max_new_tokens,