Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -66,14 +66,15 @@ def predict(
|
|
| 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 |
"""
|
| 78 |
proc_submission - a helper function for the gradio module to process submissions
|
| 79 |
|
|
@@ -356,7 +357,7 @@ if __name__ == "__main__":
|
|
| 356 |
fn=proc_submission,
|
| 357 |
inputs=[
|
| 358 |
input_text,
|
| 359 |
-
model_name,
|
| 360 |
num_beams,
|
| 361 |
token_batch_length,
|
| 362 |
length_penalty,
|
|
|
|
| 66 |
|
| 67 |
def proc_submission(
|
| 68 |
input_text: str,
|
|
|
|
| 69 |
num_beams: int =2,
|
| 70 |
token_batch_length: int =1024,
|
| 71 |
length_penalty: float = 0.7,
|
| 72 |
repetition_penalty: float =3.5,
|
| 73 |
no_repeat_ngram_size: int =3,
|
| 74 |
+
max_input_length: int = 2560):
|
| 75 |
+
model_name="pszemraj/led-large-book-summary"
|
| 76 |
+
|
| 77 |
+
|
| 78 |
"""
|
| 79 |
proc_submission - a helper function for the gradio module to process submissions
|
| 80 |
|
|
|
|
| 357 |
fn=proc_submission,
|
| 358 |
inputs=[
|
| 359 |
input_text,
|
| 360 |
+
#model_name,
|
| 361 |
num_beams,
|
| 362 |
token_batch_length,
|
| 363 |
length_penalty,
|