Update app_v2.py

app_v2.py

@@ -79,7 +79,7 @@ def caption(input_image: Image.Image, prompt: str, temperature: float, top_p: fl
     output = cap_model.generate(**generate_kwargs)
     print(f"Generated {len(output[0])} tokens")
 
-@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)