commit

app.py

@@ -545,7 +545,7 @@ def upscaler(
         tile_size=(tile_height, tile_width),
         denoise_strength=denoise_strength,
         num_inference_steps=num_inference_steps,
-        loras_scale={"more_details": 0.5, "sdxl_render": 1.0},
+        loras_scale={"more_details": 1.0, "sdxl_render": 1.0},
         solver_type=solver_type,
     )
 
@@ -554,14 +554,13 @@ def upscaler(
     return enhanced_image
 
 def summarize_text(
-    text, max_length=20, num_beams=64, early_stopping=True
+    text, max_length=10
 ):
     log(f'CALL summarize_text')
     summary = pegasus_tokenizer.decode( pegasus_model.generate(
         pegasus_tokenizer(text,return_tensors="pt").input_ids,
         max_length=max_length,
-        num_beams=num_beams,
-        early_stopping=early_stopping
+        early_stopping=True
     )[0], skip_special_tokens=True)
     log(f'RET summarize_text with summary as {summary}')
     return summary
@@ -620,7 +619,7 @@ def all_pipes(pos,neg,artist,song):
     for img in imgs:
         labeled_img = add_song_cover_text(img,artist,song)
         enhanced_img = upscaler(labeled_img)
-        name = f'{pos_artist} - {pos_song} ({index}).png'
+        name = f'{artist} - {song} ({index}).png'
         enhanced_img.save(name)
         names.append(name)
     return names