Update app.py

app.py

@@ -566,7 +566,7 @@ def _summarize(text):
     gen = model.generate(
         toks,
         length_penalty=0.1,
-        num_beams=8,
+        num_beams=4,
         early_stopping=True,
         max_length=512
     )
@@ -1278,7 +1278,7 @@ def handle_generation(artist,song,genre,lyrics):
     pos_song = re.sub(r"([ \t]){1,}", " ", song).lower().strip()
     pos_song = ' '.join(word[0].upper() + word[1:] for word in pos_song.split())
 
-    lyrics = re.sub(r"(, | ,)",". ",lyrics)
+    lyrics = re.sub(r"[,]",".",lyrics)
     lyrics_lines = re.split(r"([\n]){1,}", lyrics)
     
     for line_index in range(len(lyrics_lines)):
@@ -1291,12 +1291,13 @@ def handle_generation(artist,song,genre,lyrics):
     pos_genre = re.sub(f'[{punctuation}]', '', re.sub(r"([ \t]){1,}", " ", genre)).lower().strip()
     pos_genre = ' '.join(word[0].upper() + word[1:] for word in pos_genre.split())
 
-    pos_lyrics = re.sub(f'[{punctuation}]', '', re.sub(r"([ \t]){1,}", " ", lyrics)).lower().strip()
-    pos_lyrics_sum = pos_lyrics if pos_lyrics == "" else summarize(translate(pos_lyrics))
+    pos_lyrics = re.sub(r"([ \t]){1,}", " ", lyrics).lower().strip()
+    pos_lyrics = pos_lyrics if pos_lyrics == "" else summarize(translate(pos_lyrics))
+    pos_lyrics = re.sub(r"([ \t]){1,}", " ", re.sub(f'[{punctuation}]', '', pos_lyrics)).lower().strip()
 
     neg = f"Sexuality, Human body, Human, Textual, Labeled, Distorted, Discontinuous, Blurry, Doll-Like, Overly Plastic, Low-Quality, Painted, Smooth, Artificial, Phony, Gaudy, Digital Effects."
     q = "\""
-    pos = f'{ translate(pos_genre) } song hyper-realistic official image{ pos_lyrics_sum if pos_lyrics_sum == "" else ": " + pos_lyrics_sum }.'
+    pos = f'{ translate(pos_genre) } song hyper-realistic official image{ pos_lyrics if pos_lyrics == "" else ": " + q + pos_lyrics + q }.'
 
     print(f"""
         Positive: {pos}