Update app.py

app.py

@@ -429,13 +429,12 @@ root.addHandler(handler)
 # constant data
 
 base = "black-forest-labs/FLUX.1-schnell"
-pegasus_name = "google/pegasus-xsum"
 
 # precision data
 
 seq=256
-width=1024
-height=1024
+width=1536
+height=1536
 image_steps=8
 img_accu=0
 
@@ -509,7 +508,7 @@ def upscaler(
     prompt: str = "Photorealistic, Hyperrealistic, Realistic Photography, High-Quality Photography, Natural.",
     negative_prompt: str = "Distorted, Discontinuous, Blurry, Doll-Like, Overly-Plastic, Low-Quality, Painted, Smoothed, Artificial, Phony, Gaudy, Digital Effects.",
     seed: int = int(str(random.random()).split(".")[1]),
-    upscale_factor: int = 2,
+    upscale_factor: int = 4,
     controlnet_scale: float = 0.6,
     controlnet_decay: float = 1.0,
     condition_scale: int = 6,
@@ -539,7 +538,7 @@ def upscaler(
         tile_size=(tile_height, tile_width),
         denoise_strength=denoise_strength,
         num_inference_steps=num_inference_steps,
-        loras_scale={"more_details": 1.0, "sdxl_render": 1.0},
+        loras_scale={"more_details": 0.5, "sdxl_render": 1.0},
         solver_type=solver_type,
     )
 
@@ -548,11 +547,11 @@ def upscaler(
     return enhanced_image
 
 def summarize_text(
-    text, max_length=16
+    text, min_length=4, max_length=12
 ):
     log(f'CALL summarize_text')
-    summ = pipeline("summarization", model="google/pegasus-xsum")
-    summary = summ("summarize: " + text)[0]["summary_text"]
+    summ = pipeline("summarization", model="facebook/bart-large-cnn")
+    summary = summ("summarize: " + text, max_length=max_length, min_length=min_length, do_sample=False)[0]["summary_text"]
     log(f'RET summarize_text with summary as {summary}')
     return summary