commit

app.py

@@ -125,6 +125,14 @@ pegasus_model = PegasusForConditionalGeneration.from_pretrained(pegasus_name)
 
 # functionality
 
+def summarize_text(text, model_name="google/pegasus-xsum", max_length=30, num_beams=4, early_stopping=True):
+    return pegasus_tokenizer.decode( pegasus_model.generate(
+        pegasus_tokenizer(pos_lyrics,return_tensors="pt").input_ids,
+        max_length=max_length,
+        num_beams=num_beams,
+        early_stopping=early_stopping
+    )[0], skip_special_tokens=True)
+
 def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
@@ -150,7 +158,7 @@ def handle_generate(artist,song,genre,lyrics):
     pos_song = ' '.join(word[0].upper() + word[1:] for word in pos_song.split())
     pos_genre = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", genre)).upper().strip()
     pos_lyrics = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", lyrics)).lower().strip()
-    pos_lyrics_sum = pegasus_tokenizer.decode(pegasus_model.generate(pegasus_tokenizer(pos_lyrics,return_tensors="pt").input_ids)[0], skip_special_tokens=True)
+    pos_lyrics_sum = summarize_text(pos_lyrics)
     neg = f"Textual Labeled Distorted Discontinuous Ugly Blurry"
     pos = f'Realistic Natural Genuine Reasonable Detailed, { pos_genre } GENRE SONG - { pos_song }: "{ pos_lyrics_sum }"'