commit

app.py

@@ -13,6 +13,7 @@ import sys
 import warnings
 import time
 import asyncio
+import math
 from functools import partial
 
 # external
@@ -122,9 +123,9 @@ image_pipe.enable_model_cpu_offload()
 
 # functionality
 
-@spaces.GPU(duration=40)
+@spaces.GPU(duration=70)
 def summarize_text(
-    text, max_length=30, num_beams=4, early_stopping=True, 
+    text, max_length=30, num_beams=16, early_stopping=True, 
     pegasus_tokenizer = PegasusTokenizerFast.from_pretrained("google/pegasus-xsum"),
     pegasus_model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
 ):
@@ -139,7 +140,7 @@ def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU(duration=120)
+@spaces.GPU(duration=140)
 def pipe_generate(p1,p2):
     return image_pipe(
             prompt=p1,
@@ -175,22 +176,22 @@ def handle_generate(artist,song,genre,lyrics):
     draw = ImageDraw.Draw(img)
 
     rows = 1
-    labes_distance = 1 // 3
+    labes_distance = math.cail(1 / 3)
 
-    textheight=min(( width // 10 ), ( height // 5 ))
+    textheight=min(math.cail( width / 10 ), math.cail( height / 5 ))
     font = ImageFont.truetype(r"Alef-Bold.ttf", textheight)
     textwidth = draw.textlength(pos_song,font)
-    x = (width - textwidth) // 2
-    y = (height - (textheight * rows // 2)) // 2
-    y = y - (y // labes_distance)
+    x = math.cail((width - textwidth) / 2)
+    y = math.cail((height - math.cail(textheight * rows / 2)) / 2)
+    y = y - math.cail(y / labes_distance)
     draw.text((x, y), pos_song, (255,255,255), font=font)
 
-    textheight=min(( width // 12 ), ( height // 6 ))
+    textheight=min(math.cail( width / 12 ), math.cail( height / 6 ))
     font = ImageFont.truetype(r"Alef-Bold.ttf", textheight)
     textwidth = draw.textlength(pos_artist,font)
-    x = (width - textwidth) // 2
-    y = (height - (textheight * rows // 2)) // 2
-    y = y + (y // labes_distance)
+    x = math.cail((width - textwidth) / 2)
+    y = math.cail((height - math.cail(textheight * rows / 2)) / 2)
+    y = y + math.cail(y / labes_distance)
     draw.text((x, y), pos_artist, (255,255,255), font=font)
     
     name = generate_random_string(12) + ".png"