sagawa commited on
Commit
d7574d5
·
verified ·
1 Parent(s): fa014b1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -163,8 +163,6 @@ if st.button('predict'):
163
  with torch.no_grad():
164
  output = model.generate(
165
  **inputs,
166
- min_length=CFG.output_min_length,
167
- max_length=CFG.output_max_length,
168
  num_beams=CFG.num_beams,
169
  num_return_sequences=CFG.num_return_sequences,
170
  return_dict_in_generate=True,
 
163
  with torch.no_grad():
164
  output = model.generate(
165
  **inputs,
 
 
166
  num_beams=CFG.num_beams,
167
  num_return_sequences=CFG.num_return_sequences,
168
  return_dict_in_generate=True,