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

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,10 +16,11 @@ def disambiguate_sentence(sentence):
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)
 
23
  return results
24
 
25
 
 
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)
23
+ print(f"Decoded: {decoded}")
24
  return results
25
 
26