Update app.py

app.py

@@ -589,7 +589,7 @@ def _summarize(text):
     toks = tokenizer.encode( prefix + text, return_tensors="pt", truncation=False)
     gen = model.generate(
         toks,
-        length_penalty=0.1,
+        length_penalty=0.7,
         num_beams=16,
         early_stopping=True,
         max_length=512
@@ -598,7 +598,7 @@ def _summarize(text):
     log(f'RET _summarize with ret as {ret}')
     return ret
 
-def summarize(text, max_words=10):
+def summarize(text, max_words=8):
     log(f'CALL summarize')
     
     words = text.split()
@@ -1303,8 +1303,8 @@ def handle_generation(h,w,d):
             
             for line_index in range(len(d_lines)):
                 d_lines[line_index] = d_lines[line_index].strip()
-                if re.sub(r'[\.]$', '', d_lines[line_index]) == d_lines[line_index]:
-                    d_lines[line_index] = d_lines[line_index].strip() + "."
+                if d_lines[line_index] != "" and re.sub(r'[\.]$', '', d_lines[line_index]) == d_lines[line_index]:
+                    d_lines[line_index] = d_lines[line_index] + "."
             d = " ".join(d_lines)
         
             d = re.sub(r"([ \t]){1,}", " ", d).lower().strip()
@@ -1314,11 +1314,11 @@ def handle_generation(h,w,d):
                 d = d if d == "" else translate(d)
             d = re.sub(r"([ \t]){1,}", " ", d)
             d = re.sub(r"(\. \.)", ".", d)
-            d = re.sub(r"[,]", ";", d).lower().strip()
+            d = re.sub(r"[,]", ".", d).lower().strip()
     
         neg = f"Textual, Text, Blurry, Distorted, Exceptional, Irregular, Unusual, Shiny, Smoothed, Polished, Low Quality, Worst Quality, Normal Quality, Anime Quality, Paint Quality, Movie Quality."
         q = "\""
-        pos = f'Realistic Real-Life Life-Like Convincing{ "." if d == "" else ": " + d }'
+        pos = f'Convincing Realism{ "." if d == "" else " from " + d }'
     
         print(f"""
             Positive: {pos}