Update app.py

app.py

@@ -557,15 +557,18 @@ def get_tensor_length(tensor):
 def summarize(
     text, max_len=20, min_len=10
 ):
-    log(f'CALL summarize_text')
+    log(f'CALL summarize')
     inputs = tokenizer.encode("summarize: " + text, return_tensors="pt", max_length=float('inf'), truncation=False)
+    if get_tensor_length(inputs) < 3:
+        print("Summarization Error: Text is too short, 3 words minimum!")
+        return text
     i = 1
     while get_tensor_length(inputs) > max_len:
-        print(f'DBG summarize_text 1 {i}')
+        print(f'DBG summarize 1 {i}')
         outputs = model.generate(
-            inputs[0][:512],
+            torch.tensor(list(inputs[0][:512])),
             length_penalty=2.0,
-            num_beams=max(8,get_tensor_length(inputs)),
+            num_beams=min(4,get_tensor_length(inputs) - 1),
             early_stopping=True,
             max_length=max( get_tensor_length(inputs) // 4 , max_len ),
             min_length=min_len
@@ -573,7 +576,7 @@ def summarize(
         inputs = torch.tensor([[*list(outputs[0]), *list(inputs[0][512:])]])
         i = i + 1
     summary = tokenizer.decode(inputs[0])
-    log(f'RET summarize_text with summary as {summary}')
+    log(f'RET summarize with summary as {summary}')
     return summary
 
 def generate_random_string(length):
@@ -662,11 +665,11 @@ def handle_generation(artist,song,genre,lyrics):
     pos_genre = ' '.join(word[0].upper() + word[1:] for word in pos_genre.split())
 
     pos_lyrics = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", lyrics)).lower().strip()
-    pos_lyrics_sum = pos_lyrics if pos_lyrics == "" else summarize(pos_lyrics)
+    pos_lyrics_sum = pos_lyrics if pos_lyrics == "" else summarize(translate(pos_lyrics))
 
     neg = f"Sexuality, Humanity, Textual, Labeled, Distorted, Discontinuous, Blurry, Doll-Like, Overly Plastic, Low-Quality, Painted, Smoothed, Artificial, Phony, Gaudy, Digital Effects."
     q = "\""
-    pos = f'HQ Hyper-realistic { translate(pos_genre) } song "{ translate(pos_song) }"{ pos_lyrics_sum if pos_lyrics_sum == "" else ": " + translate(pos_lyrics_sum) }.'
+    pos = f'HQ Hyper-realistic { translate(pos_genre) } song "{ translate(pos_song) }"{ pos_lyrics_sum if pos_lyrics_sum == "" else ": " + pos_lyrics_sum }.'
 
     print(f"""
         Positive: {pos}