BigSalmon commited on
Commit
a5cd868
·
1 Parent(s): 15f3d13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -85,6 +85,7 @@ if st.button("Run"):
85
  model_name=model_name,
86
  max_len=int(max_len),
87
  temp=temp,
 
88
  top_k=int(top_k),
89
  top_p=float(top_p))
90
  print("result:", result)
 
85
  model_name=model_name,
86
  max_len=int(max_len),
87
  temp=temp,
88
+ num_return_sequences = int(num_return_sequences),
89
  top_k=int(top_k),
90
  top_p=float(top_p))
91
  print("result:", result)