ameerazam08 commited on
Commit
27d1cb4
·
verified ·
1 Parent(s): ec6c037

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ pipeline = FluxPipeline.from_pretrained(
14
  ).to('cuda')
15
 
16
 
17
- @spaces.GPU()
18
  def generate_image(image, prompt, guidance_scale, num_steps, lora_name):
19
  # Load the model
20
 
 
14
  ).to('cuda')
15
 
16
 
17
+ @spaces.GPU
18
  def generate_image(image, prompt, guidance_scale, num_steps, lora_name):
19
  # Load the model
20