Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def proc_submission(
|
|
24 |
summary_type: str,
|
25 |
num_beams,
|
26 |
token_batch_length,
|
27 |
-
|
28 |
#repetition_penalty,
|
29 |
#no_repeat_ngram_size: int = 3,
|
30 |
max_input_length: int = 768,
|
@@ -55,7 +55,6 @@ def proc_submission(
|
|
55 |
"min_length": 11,
|
56 |
"max_length": int(token_batch_length // 4),
|
57 |
"early_stopping": True,
|
58 |
-
#"do_sample": False,
|
59 |
}
|
60 |
st = time.perf_counter()
|
61 |
history = {}
|
|
|
24 |
summary_type: str,
|
25 |
num_beams,
|
26 |
token_batch_length,
|
27 |
+
length_penalty,
|
28 |
#repetition_penalty,
|
29 |
#no_repeat_ngram_size: int = 3,
|
30 |
max_input_length: int = 768,
|
|
|
55 |
"min_length": 11,
|
56 |
"max_length": int(token_batch_length // 4),
|
57 |
"early_stopping": True,
|
|
|
58 |
}
|
59 |
st = time.perf_counter()
|
60 |
history = {}
|