Update app.py

app.py

@@ -51,8 +51,8 @@ from datetime import datetime
 
 working = False
 
-model = T5ForConditionalGeneration.from_pretrained("t5-base")
-tokenizer = T5Tokenizer.from_pretrained("t5-base")
+model = T5ForConditionalGeneration.from_pretrained("t5-large")
+tokenizer = T5Tokenizer.from_pretrained("t5-large")
 
 def log(msg):
     print(f'{datetime.now().time()} {msg}')
@@ -571,8 +571,8 @@ def _summarize(text):
     toks = tokenizer.encode( prefix + text, return_tensors="pt", truncation=False)
     gen = model.generate(
         toks,
-        length_penalty=0.1,
-        num_beams=4,
+        length_penalty=0.01,
+        num_beams=8,
         early_stopping=True,
         max_length=512
     )
@@ -598,13 +598,19 @@ def summarize(text, max_words=10):
     if words_length >= 510:
         while words_length >= 510:
             words = text.split()
-            text = _summarize(
+            sum = _summarize(
                 " ".join(words[0:510])
             ) + " ".join(words[510:])
+            if summ == text:
+                return text
+            text = summ
             words_length = len(text.split())
     
     while words_length > max_words:
-        text = _summarize(text)
+        summ = _summarize(text)
+        if summ == text:
+            return text
+        text = summ
         words_length = len(text.split())
         
     log(f'RET summarize with text as {text}')
@@ -1370,7 +1376,7 @@ if __name__ == "__main__":
                 width = gr.Slider(
                     label="Width (px)",
                     minimum=512,
-                    maximum=1536,
+                    maximum=1024,
                     step=64,
                     value=1024,
                 )
@@ -1379,7 +1385,7 @@ if __name__ == "__main__":
                 height = gr.Slider(
                     label="Height (px)",
                     minimum=512,
-                    maximum=1536,
+                    maximum=1024,
                     step=64,
                     value=1024,
                 )