Spaces:
Runtime error
Runtime error
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 |
"""
|