main

gradio_app.py

@@ -106,6 +106,7 @@ class ResizeWithPadding:
         return img
 
 # The function to generate image from a prompt and conditional image
+@spaces.GPU(duration=180)
 def infer(prompt, sample_image, frame_num, seed=0, randomize_seed=False):
     logger.info(f"Started generating image with prompt: {prompt}")