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