Spaces:
Sleeping
Sleeping
commit
Browse files
app.py
CHANGED
@@ -434,8 +434,8 @@ pegasus_name = "google/pegasus-xsum"
|
|
434 |
# precision data
|
435 |
|
436 |
seq=512
|
437 |
-
width=
|
438 |
-
height=
|
439 |
image_steps=8
|
440 |
img_accu=0
|
441 |
|
@@ -554,7 +554,7 @@ def upscaler(
|
|
554 |
return enhanced_image
|
555 |
|
556 |
def summarize_text(
|
557 |
-
text, max_length=
|
558 |
):
|
559 |
log(f'CALL summarize_text')
|
560 |
summary = pegasus_tokenizer.decode( pegasus_model.generate(
|
|
|
434 |
# precision data
|
435 |
|
436 |
seq=512
|
437 |
+
width=768
|
438 |
+
height=768
|
439 |
image_steps=8
|
440 |
img_accu=0
|
441 |
|
|
|
554 |
return enhanced_image
|
555 |
|
556 |
def summarize_text(
|
557 |
+
text, max_length=20, num_beams=64, early_stopping=True
|
558 |
):
|
559 |
log(f'CALL summarize_text')
|
560 |
summary = pegasus_tokenizer.decode( pegasus_model.generate(
|