zmbfeng commited on
Commit
fd4136c
·
verified ·
1 Parent(s): 128b743

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -20,7 +20,9 @@ def create_response_original(input_str,
20
  top_k,
21
  do_sample):
22
  print("input_str="+input_str)
 
23
  print("num_beams=" + str(num_beams))
 
24
  print("num_return_sequences" + str(num_return_sequences))
25
  print("top_p" + str(top_p))
26
  print("top_k" + str(top_k))
 
20
  top_k,
21
  do_sample):
22
  print("input_str="+input_str)
23
+ num_beams = int(num_beams)
24
  print("num_beams=" + str(num_beams))
25
+ num_return_sequences=int(num_return_sequences)
26
  print("num_return_sequences" + str(num_return_sequences))
27
  print("top_p" + str(top_p))
28
  print("top_k" + str(top_k))