Update app.py

app.py

@@ -451,8 +451,8 @@ MAX_SEED = np.iinfo(np.int32).max
 # precision data
 
 seq=512
-image_steps=30
-img_accu=4.5
+image_steps=50
+img_accu=8.5
 
 # ui data
 
@@ -528,7 +528,7 @@ torch.cuda.empty_cache()
 def upscaler(
     input_image: Image.Image,
     prompt: str = "Masterpiece, Best Quality, Hyper-Realistic, Super-Realistic, Natural, Reasonable, Logical.",
-    negative_prompt: str = "Blurry, Distorted, Exceptional, Irregular, Unusual, Shiny, Smoothed, Polished, Low Quality, Worst Quality, Normal Quality, Anime Quality, Movies Quality.",
+    negative_prompt: str = "Blurry, Distorted, Exceptional, Irregular, Unusual, Shiny, Smoothed, Polished, Low Quality, Worst Quality, Normal Quality, Anime Quality, Paint Quality, Movie Quality.",
     seed: int = random.randint(0, MAX_SEED),
     upscale_factor: int = 2,
     controlnet_scale: float = 0.6,
@@ -594,8 +594,8 @@ def _summarize(text):
     toks = tokenizer.encode( prefix + text, return_tensors="pt", truncation=False)
     gen = model.generate(
         toks,
-        length_penalty=2.0,
-        num_beams=8,
+        length_penalty=0.1,
+        num_beams=16,
         early_stopping=True,
         max_length=512
     )
@@ -603,7 +603,7 @@ def _summarize(text):
     log(f'RET _summarize with ret as {ret}')
     return ret
 
-def summarize(text, max_len=400):
+def summarize(text, max_words=20):
     log(f'CALL summarize')
     
     words = text.split()
@@ -620,11 +620,12 @@ def summarize(text, max_len=400):
             text = summ
             words_length = len(text.split())
     
-    while len(text) > max_len:
+    while words_length > max_words:
         summ = _summarize(text)
         if summ == text:
             return text
         text = summ
+        words_length = len(text.split())
         
     log(f'RET summarize with text as {text}')
     return text
@@ -1319,9 +1320,9 @@ def handle_generation(h,w,d):
             d = re.sub(r"([ \t]){1,}", " ", d)
             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, Movies Quality."
+        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'Masterpiece, Best Quality, Hyper-Realistic, Super-Realistic, Natural, Reasonable, Logical, { d if d == "" else ", " + d }'
+        pos = f'Masterpiece, Best Quality, Hyper-Realistic, Super-Realistic, Natural, Reasonable, Logical{ "." if d == "" else ". " + d }'
     
         print(f"""
             Positive: {pos}