yaron123 commited on
Commit
459972b
·
1 Parent(s): 75961cb
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -434,8 +434,8 @@ pegasus_name = "google/pegasus-xsum"
434
  # precision data
435
 
436
  seq=512
437
- width=1536
438
- height=1536
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=10, num_beams=4, early_stopping=True
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(