Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -66,12 +66,12 @@ def predict(
|
|
66 |
|
67 |
def proc_submission(
|
68 |
input_text: str,
|
69 |
-
model_name: str,
|
70 |
-
num_beams: int,
|
71 |
-
token_batch_length: int,
|
72 |
-
length_penalty: float,
|
73 |
-
repetition_penalty: float,
|
74 |
-
no_repeat_ngram_size: int,
|
75 |
max_input_length: int = 2560,
|
76 |
):
|
77 |
"""
|
|
|
66 |
|
67 |
def proc_submission(
|
68 |
input_text: str,
|
69 |
+
model_name: str ="pszemraj/led-large-book-summary",
|
70 |
+
num_beams: int =2,
|
71 |
+
token_batch_length: int =1024,
|
72 |
+
length_penalty: float = 0.7,
|
73 |
+
repetition_penalty: float =3.5,
|
74 |
+
no_repeat_ngram_size: int =3,
|
75 |
max_input_length: int = 2560,
|
76 |
):
|
77 |
"""
|