Update app_v2.py

app_v2.py

@@ -41,7 +41,7 @@ pipe = FluxControlNetPipeline.from_pretrained(
 )
 pipe.to("cuda")
 
-@spaces.GPU()
+@spaces.GPU(duration=10)
 @torch.no_grad()
 def caption(input_image: Image.Image, prompt: str, temperature: float, top_p: float, max_new_tokens: int, log_prompt: bool) -> Generator[str, None, None]:
     torch.cuda.empty_cache()
@@ -85,7 +85,7 @@ def caption(input_image: Image.Image, prompt: str, temperature: float, top_p: fl
 
     #return output[0]
 
-@spaces.GPU()
+@spaces.GPU(duration=10)
 @torch.no_grad()
 def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale, seed, guidance_end):
     generator = torch.Generator().manual_seed(seed)