Update README.md
Browse files
README.md
CHANGED
@@ -24,8 +24,32 @@ It achieves the following results on the evaluation set:
|
|
24 |
|
25 |
## Model description
|
26 |
|
27 |
-
|
28 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
29 |
## Intended uses & limitations
|
30 |
|
31 |
More information needed
|
|
|
24 |
|
25 |
## Model description
|
26 |
|
27 |
+
for i in range(0,14):
|
28 |
+
input_ids = tokenizer(
|
29 |
+
[WHITESPACE_HANDLER(sumariosDuplos[i])],
|
30 |
+
return_tensors="pt",
|
31 |
+
padding="max_length",
|
32 |
+
truncation=True,
|
33 |
+
max_length=512
|
34 |
+
)["input_ids"]
|
35 |
+
|
36 |
+
output_ids = model.generate(
|
37 |
+
input_ids=input_ids,
|
38 |
+
max_length=200,
|
39 |
+
min_length=75,
|
40 |
+
no_repeat_ngram_size=2,
|
41 |
+
num_beams=5
|
42 |
+
)[0]
|
43 |
+
|
44 |
+
summary = tokenizer.decode(
|
45 |
+
output_ids,
|
46 |
+
skip_special_tokens=True,
|
47 |
+
clean_up_tokenization_spaces=False
|
48 |
+
)
|
49 |
+
|
50 |
+
sumariosFinal.append(summary)
|
51 |
+
print(i,"\n",summary,"\n")
|
52 |
+
|
53 |
## Intended uses & limitations
|
54 |
|
55 |
More information needed
|