commit

app.py

@@ -120,12 +120,14 @@ function custom(){
 image_pipe = FluxPipeline.from_pretrained(base, torch_dtype=torch.bfloat16).to(device)
 image_pipe.enable_model_cpu_offload()
 
-pegasus_tokenizer = PegasusTokenizerFast.from_pretrained(pegasus_name)
-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):
+@spaces.GPU(duration=40)
+def summarize_text(
+    text, max_length=60, num_beams=4, early_stopping=True, 
+    pegasus_tokenizer = PegasusTokenizerFast.from_pretrained("google/pegasus-xsum"),
+    pegasus_model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
+):
     return pegasus_tokenizer.decode( pegasus_model.generate(
         pegasus_tokenizer(text,return_tensors="pt").input_ids,
         max_length=max_length,
@@ -137,7 +139,7 @@ def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU()
+@spaces.GPU(duration=80)
 def pipe_generate(p1,p2):
     return image_pipe(
             prompt=p1,
@@ -160,7 +162,7 @@ def handle_generate(artist,song,genre,lyrics):
     pos_lyrics = re.sub(f'[{punctuation}]', '', re.sub("([ \t\n]){1,}", " ", lyrics)).lower().strip()
     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 }"'
+    pos = f'Realistic Natural Genuine Reasonable Detailed { pos_genre } GENRE { pos_song } "{ pos_lyrics_sum }"'
 
     print(f"""
         Positive: {pos}