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,
|