Upload code

app.py

@@ -111,14 +111,14 @@ def resize_without_crop(image, target_width, target_height):
 
 
 @torch.inference_mode()
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=360)
 def interrogator_process(x):
     image_description = wd14tagger.default_interrogator(x)
     return image_description, image_description
 
 
 @torch.inference_mode()
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=360)
 def process(input_fg, prompt, input_undo_steps, image_width, image_height, seed, steps, n_prompt, cfg,
             progress=gr.Progress()):
     rng = torch.Generator(device=memory_management.gpu).manual_seed(int(seed))
@@ -215,7 +215,7 @@ def process_video_inner(image_1, image_2, prompt, seed=123, steps=25, cfg_scale=
 
 
 @torch.inference_mode()
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=360)
 def process_video(keyframes, prompt, steps, cfg, fps, seed, progress=gr.Progress()):
     result_frames = []
     cropped_images = []