commit

app.py

@@ -557,7 +557,7 @@ def summarize_text(
     summary = pegasus_tokenizer.decode( pegasus_model.generate(
         pegasus_tokenizer(text,return_tensors="pt").input_ids,
         max_length=max_length,
-        num_beams = 2,
+        num_beams = 4,
         early_stopping=True
     )[0], skip_special_tokens=True)
     log(f'RET summarize_text with summary as {summary}')
@@ -631,6 +631,7 @@ def handle_generation(artist,song,genre,lyrics):
 
     pos_lyrics = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", lyrics)).lower().strip()
     pos_lyrics_sum = pos_lyrics if pos_lyrics == "" else summarize_text(pos_lyrics)
+
     neg = f"Sexual, Textual, Labeled, Distorted, Discontinuous, Blurry, Doll-Like, Overly-Plastic, Low-Quality, Painted, Smoothed, Artificial, Phony, Gaudy, Digital Effects."
     q = "\""
     pos = f'Photorealistic, Hyperrealistic, Realistic Photography, High-Quality Photography, Natural, made for the { pos_genre } song "{ pos_song }"{ pos_lyrics_sum if pos_lyrics_sum == "" else ": " + q + pos_lyrics_sum + q }.'
@@ -646,7 +647,8 @@ def handle_generation(artist,song,genre,lyrics):
     index = 1
     names = []
     for img in imgs:
-        labeled_img = add_song_cover_text(img,artist,song,height*4,width*4)
+        scaled_by = 2
+        labeled_img = add_song_cover_text(img,artist,song,height*scaled_by,width*scaled_by)
         name = f'{artist} - {song} ({index}).png'
         labeled_img.save(name)
         names.append(name)