Kongfha commited on
Commit
135eff8
·
1 Parent(s): 9225de7

debug int temperature

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def generate(input_sentence, top_k=50, temperature=1.0, max_length=140):
14
  max_length=int(max_length),
15
  do_sample=True,
16
  top_k=int(top_k),
17
- temperature=temperature)
18
  return generated_text[0]['generated_text']
19
 
20
  inputs = [
 
14
  max_length=int(max_length),
15
  do_sample=True,
16
  top_k=int(top_k),
17
+ temperature=float(temperature))
18
  return generated_text[0]['generated_text']
19
 
20
  inputs = [