Update app.py

app.py

@@ -565,16 +565,16 @@ def _summarize(text):
     toks = tokenizer.encode( prefix + text, return_tensors="pt", truncation=False)
     gen = model.generate(
         toks,
-        length_penalty=3.0,
-        num_beams=math.ceil(len(text.split()) / 6),
+        length_penalty=0.1,
+        num_beams=16,
         early_stopping=True,
-        max_length=math.ceil(len(text.split()) / 4)
+        max_length=512
     )
     ret = tokenizer.decode(gen[0], skip_special_tokens=True)
     log(f'RET _summarize with ret as {ret}')
     return ret
 
-def summarize(text, max_words=20):
+def summarize(text, max_words=10):
     log(f'CALL summarize')
     
     words = text.split()
@@ -590,12 +590,10 @@ def summarize(text, max_words=20):
     words_length = len(text.split())
 
     if words_length >= 510:
-        shrink_step = 500 // (len(text.split()) / 500)
         while words_length >= 510:
             words = text.split()
-            above_limit = words[510:]
             text = _summarize(
-                " ".join(words[0:510]), shrink_step
+                " ".join(words[0:510])
             ) + " ".join(words[510:])
             words_length = len(text.split())
     
@@ -1296,9 +1294,9 @@ def handle_generation(artist,song,genre,lyrics):
     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))
 
-    neg = f"Sexuality, Humanity, Textual, Labeled, Distorted, Discontinuous, Blurry, Doll-Like, Overly Plastic, Low-Quality, Painted, Smoothed, Artificial, Phony, Gaudy, Digital Effects."
+    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'HQ Hyper-realistic { translate(pos_genre) } song "{ translate(pos_song) }"{ pos_lyrics_sum if pos_lyrics_sum == "" else ": " + pos_lyrics_sum }.'
+    pos = f'{ translate(pos_genre) } song hyper-realistic official image{ pos_lyrics_sum if pos_lyrics_sum == "" else ": " + pos_lyrics_sum }.'
 
     print(f"""
         Positive: {pos}