commit

app.py

@@ -124,7 +124,7 @@ image_pipe.enable_model_cpu_offload()
 
 @spaces.GPU(duration=40)
 def summarize_text(
-    text, max_length=60, num_beams=4, early_stopping=True, 
+    text, max_length=30, num_beams=4, early_stopping=True, 
     pegasus_tokenizer = PegasusTokenizerFast.from_pretrained("google/pegasus-xsum"),
     pegasus_model = PegasusForConditionalGeneration.from_pretrained("google/pegasus-xsum")
 ):
@@ -139,7 +139,7 @@ def generate_random_string(length):
     characters = str(ascii_letters + digits)
     return ''.join(random.choice(characters) for _ in range(length))
 
-@spaces.GPU(duration=80)
+@spaces.GPU(duration=120)
 def pipe_generate(p1,p2):
     return image_pipe(
             prompt=p1,