Commit
Β·
001bca6
1
Parent(s):
94113a9
update data
Browse files
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)
|