Update app.py

app.py

@@ -51,7 +51,7 @@ def get_prompt(file:str):
         a=f.readlines()
     return a #a[0]:positive prompt, a[1] negative prompt
 
-@spaces.GPU
+@spaces.GPU(duration=360)
 def init_pipe():
     def unwarp_model(state_dict):
         new_state_dict = {}
@@ -126,7 +126,8 @@ def init_pipe():
     pipe.enable_model_cpu_offload()
 
     return pipe
-@spaces.GPU
+    
+@spaces.GPU(duration=360)
 def inference(source_images, 
         target_images, 
         text_prompt, negative_prompt,