Akjava commited on
Commit
546800a
·
1 Parent(s): 635bdcc
Files changed (2) hide show
  1. app.py +3 -1
  2. flux1_inpaint.py +1 -1
app.py CHANGED
@@ -15,8 +15,10 @@ def sanitize_prompt(prompt):
15
  return sanitized_prompt
16
 
17
  @spaces.GPU(duration=180)
18
- def process_images(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0):
19
  # I'm not sure when this happen
 
 
20
  if not isinstance(image, dict):
21
  if image2 == None:
22
  print("empty mask")
 
15
  return sanitized_prompt
16
 
17
  @spaces.GPU(duration=180)
18
+ 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)):
19
  # I'm not sure when this happen
20
+ progress(0, desc="start-process-images")
21
+ print("start-process-images")
22
  if not isinstance(image, dict):
23
  if image2 == None:
24
  print("empty mask")
flux1_inpaint.py CHANGED
@@ -7,7 +7,7 @@ import spaces
7
 
8
  # I only test with FLUX.1-schnell
9
 
10
- @spaces.GPU
11
  def process_image(image,mask_image,prompt="a person",model_id="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,num_inference_steps=4):
12
  if image == None:
13
  return None
 
7
 
8
  # I only test with FLUX.1-schnell
9
 
10
+ #@spaces.GPU
11
  def process_image(image,mask_image,prompt="a person",model_id="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,num_inference_steps=4):
12
  if image == None:
13
  return None