Spaces:
Running
Running
Update app.py
Browse files
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,
|