Spaces:
Sleeping
Sleeping
Quintino Fernandes
commited on
Commit
·
542722d
1
Parent(s):
4902a85
Model parameters change
Browse files- models/summarization.py +6 -6
models/summarization.py
CHANGED
@@ -12,18 +12,18 @@ class SummarizationModel:
|
|
12 |
# Model and tokenization parameters
|
13 |
inputs = self.tokenizer.encode(
|
14 |
text,
|
15 |
-
max_length=
|
16 |
truncation=True,
|
17 |
return_tensors='pt'
|
18 |
).to(self.device)
|
19 |
|
20 |
summary_ids = self.model.generate(
|
21 |
inputs,
|
22 |
-
max_length=
|
23 |
-
min_length=
|
24 |
-
num_beams=
|
25 |
-
no_repeat_ngram_size=
|
26 |
-
early_stopping=
|
27 |
)
|
28 |
|
29 |
return self.tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
|
|
12 |
# Model and tokenization parameters
|
13 |
inputs = self.tokenizer.encode(
|
14 |
text,
|
15 |
+
max_length=1024,
|
16 |
truncation=True,
|
17 |
return_tensors='pt'
|
18 |
).to(self.device)
|
19 |
|
20 |
summary_ids = self.model.generate(
|
21 |
inputs,
|
22 |
+
max_length=512,
|
23 |
+
min_length=128,
|
24 |
+
num_beams=3,
|
25 |
+
no_repeat_ngram_size=4,
|
26 |
+
early_stopping=False,
|
27 |
)
|
28 |
|
29 |
return self.tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|