Update app.py

app.py

@@ -88,7 +88,7 @@ def encode_prompt(prompt_batch, text_encoder, tokenizer, proportion_empty_prompt
 
     return prompt_embeds, prompt_masks
 
-@spaces.GPU(duration=20)
+
 @torch.no_grad()
 def model_main(args, master_port, rank, request_queue, response_queue, mp_barrier):
     # import here to avoid huggingface Tokenizer parallelism warnings
@@ -551,7 +551,7 @@ def main():
                 label="Examples",
             )  # noqa
 
-        
+        @spaces.GPU(duration=200)
         def on_submit(*args):
             for q in request_queues:
                 q.put(args)