Akjava commited on
Commit
1779015
1 Parent(s): 096c5af
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ def sanitize_prompt(prompt):
25
  sanitized_prompt = allowed_chars.sub("", prompt)
26
  return sanitized_prompt
27
 
28
- @spaces.GPU(duration=180)
29
  def process_images(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
30
  # I'm not sure when this happen
31
  progress(0, desc="start-process-images")
@@ -56,7 +56,7 @@ def process_images(image, image2=None,prompt="a girl",inpaint_model="black-fores
56
  generators = []
57
  generator = torch.Generator("cuda").manual_seed(seed)
58
  generators.append(generator)
59
- # more parameter see https://huggingface.co/docs/diffusers/api/pipelines/flux#diffusers.FluxInpaintPipeline
60
  output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,
61
  guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
62
 
 
25
  sanitized_prompt = allowed_chars.sub("", prompt)
26
  return sanitized_prompt
27
 
28
+ @spaces.GPU(duration=120)
29
  def process_images(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
30
  # I'm not sure when this happen
31
  progress(0, desc="start-process-images")
 
56
  generators = []
57
  generator = torch.Generator("cuda").manual_seed(seed)
58
  generators.append(generator)
59
+
60
  output = pipe(prompt=prompt, image=image, mask_image=mask_image,generator=generator,strength=strength,
61
  guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
62