kadirnar commited on
Commit
eea1716
·
verified ·
1 Parent(s): 9607b53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def initialize_auraflow_pipeline():
12
  ).to("cuda")
13
  return pipeline
14
 
15
- @spaces.GPU(duration=80)
16
  def generate_image(pipeline, prompt, width, height, num_inference_steps, seed, guidance_scale):
17
  """Generate an image using the AuraFlowPipeline."""
18
  generator = torch.Generator().manual_seed(seed)
 
12
  ).to("cuda")
13
  return pipeline
14
 
15
+ @spaces.GPU(duration=95)
16
  def generate_image(pipeline, prompt, width, height, num_inference_steps, seed, guidance_scale):
17
  """Generate an image using the AuraFlowPipeline."""
18
  generator = torch.Generator().manual_seed(seed)