rjiang12 commited on
Commit
00d3b24
·
1 Parent(s): 70ea1d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -18,13 +18,13 @@ def func(sentence, max_length, temperature):
18
  output_list = model.generate(
19
  input_ids,
20
  do_sample=True,
21
- max_length=max_length,
22
- temperature=temperature,
23
  top_k=50,
24
  top_p=0.95,
25
  num_return_sequences=5
26
  )
27
- output_strs = [output for output in output_list]
28
  return output_strs
29
 
30
 
 
18
  output_list = model.generate(
19
  input_ids,
20
  do_sample=True,
21
+ max_length=float(max_length),
22
+ temperature=float(temperature),
23
  top_k=50,
24
  top_p=0.95,
25
  num_return_sequences=5
26
  )
27
+ output_strs = [tokenizer.decode(output, skip_special_tokens=True) for output in output_list]
28
  return output_strs
29
 
30