emanuelaboros commited on
Commit
001bca6
Β·
1 Parent(s): 94113a9

update data

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -15,7 +15,8 @@ def disambiguate_sentence(sentence):
15
  outputs = model.generate(
16
  **tokenizer([sentence], return_tensors="pt"),
17
  num_beams=5,
18
- num_return_sequences=5
 
19
  )
20
  decoded = tokenizer.batch_decode(outputs, skip_special_tokens=True)
21
  results.append(decoded)
 
15
  outputs = model.generate(
16
  **tokenizer([sentence], return_tensors="pt"),
17
  num_beams=5,
18
+ num_return_sequences=5,
19
+ max_new_tokens=30
20
  )
21
  decoded = tokenizer.batch_decode(outputs, skip_special_tokens=True)
22
  results.append(decoded)