Spaces:
Running
on
Zero
Running
on
Zero
update
Browse files- app.py +3 -1
- 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 |
-
|
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
|