Spaces:
Running
Running
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,
|