Hugging Face's logo

Spaces:
FlowChef
/
FlowChef-Flux1-dev
Running on Zero

App Files Community
mpatel57 commited on
Commit
e5caad5
·
verified ·
1 Parent(s): c3a398d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=
19
  pipe.to("cuda") # Comment this line if GPU is not available
20
 
21
  # Function to process the image
22
- @spaces.GPU(duration=120)
23
  def process_image(
24
  mode, image_layers, prompt, edit_prompt, seed, randomize_seed, num_inference_steps,
25
  max_steps, learning_rate, max_source_steps, optimization_steps, true_cfg, mask_input
 
19
  pipe.to("cuda") # Comment this line if GPU is not available
20
 
21
  # Function to process the image
22
+ @spaces.GPU(duration=90)
23
  def process_image(
24
  mode, image_layers, prompt, edit_prompt, seed, randomize_seed, num_inference_steps,
25
  max_steps, learning_rate, max_source_steps, optimization_steps, true_cfg, mask_input