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

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=100)
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=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)