commit

app.py

@@ -509,8 +509,8 @@ pegasus_model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-
 
 def upscaler(
     input_image: Image.Image,
-    prompt: str = "Best-Quality Realistic Genuine Reasonable Highly-Detailed",
-    negative_prompt: str = "Distorted Discontinuous Ugly Blurry Low-Quality Worst-Quality Normal-Quality Low-Resolution Painted",
+    prompt: str = "Photorealistic, Hyperrealistic, Realistic Photography, High-Quality Photography, Natural.",
+    negative_prompt: str = "Distorted, Discontinuous, Blurry, Doll-Like, Overly-Plastic, Low-Quality, Painted, Smoothed, Artificial, Phony, Gaudy, Digital Effects.",
     seed: int = int(str(random.random()).split(".")[1]),
     upscale_factor: int = 4,
     controlnet_scale: float = 0.6,
@@ -584,7 +584,7 @@ def pipe_generate_image(p1,p2):
 
 def add_song_cover_text(img,artist,song,height,width):
 
-    draw = ImageDraw.Draw(img)
+    draw = ImageDraw.Draw(img,mode="RGBA")
 
     rows = 1
     labels_distance = 1/3
@@ -607,7 +607,7 @@ def add_song_cover_text(img,artist,song,height,width):
 
     return img
 
-@spaces.GPU(duration=300)
+@spaces.GPU(duration=180)
 def all_pipes(pos,neg,artist,song):
     imgs = pipe_generate_image(pos,neg)
 
@@ -631,9 +631,9 @@ def handle_generate(artist,song,genre,lyrics):
     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 = pos_lyrics if pos_lyrics == "" else summarize_text(pos_lyrics)
-    neg = f"Textual Labeled Distorted Discontinuous Ugly Blurry Low-Quality Worst-Quality Low-Resolution Painted Smoothed"
+    neg = f"Sexual, Textual, Labeled, Distorted, Discontinuous, Blurry, Doll-Like, Overly-Plastic, Low-Quality, Painted, Smoothed, Artificial, Phony, Gaudy, Digital Effects."
     q = "\""
-    pos = f'Rough Realistic Natural Genuine Reasonable Highly-Detailed { pos_genre } MUSIC GENRE { pos_song }{ pos_lyrics_sum if pos_lyrics_sum == "" else " " + q + pos_lyrics_sum + 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 }.'
 
     print(f"""
         Positive: {pos}