change py

app.py

@@ -27,7 +27,7 @@ def sanitize_prompt(prompt):
 
 
 
-@spaces.GPU(duration=160)
+@spaces.GPU(duration=120)
 def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,inference_step=4,progress=gr.Progress(track_tqdm=True)):
     print("start process_images")
     progress(0, desc="Starting")
@@ -35,7 +35,7 @@ def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,infer
     if not isinstance(image, dict):
         if image2 == None:
             print("empty mask")
-            return image
+            return image,None
         else:
             image = dict({'background': image, 'layers': [image2]})
 
@@ -62,7 +62,7 @@ def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,infer
         # more parameter see https://huggingface.co/docs/diffusers/api/pipelines/flux#diffusers.FluxInpaintPipeline
         print(prompt)
         output = pipe(prompt=prompt, image=image,generator=generator,strength=strength
-                    ,guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=512)
+                    ,guidance_scale=0,num_inference_steps=num_inference_steps,max_sequence_length=256)
 
         # TODO support mask
         return output.images[0]
@@ -70,7 +70,7 @@ def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,infer
     output = process_img2img(image["background"],mask,prompt,strength,seed,inference_step)
    
     print("end process_images")
-    return output
+    return output,mask
     
 
 def read_file(path: str) -> str:
@@ -79,15 +79,6 @@ def read_file(path: str) -> str:
 
     return content
 
-def example_out(image,image_mask,prompt,strength,example_id):
-    # input
-    #parent,file=os.path.split(image_mask) # image is complex dict
-    #base,ext = os.path.splitext(file)
-    #key = base.split("_")[0]
-    return f"images/{example_id}.jpg"
-    #loaded_image =  Image.open(f"images/{example_id}.jpg")
-    #return loaded_image
-    #return np.array(loaded_image)
 
 css="""
 #col-left {
@@ -116,7 +107,7 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
                     with gr.Accordion(label="Advanced Settings", open=False):
                         with gr.Row( equal_height=True):
                             strength = gr.Number(value=0.75, minimum=0, maximum=0.75, step=0.01, label="strength")
-                            seed = gr.Number(value=0, minimum=0, step=1, label="seed")
+                            seed = gr.Number(value=100, minimum=0, step=1, label="seed")
                             inference_step = gr.Number(value=4, minimum=1, step=4, label="inference_step")
                         #models = ["black-forest-labs/FLUX.1-schnell"]
                         #inpaint_model = gr.Dropdown(label="modes", choices=models, value="black-forest-labs/FLUX.1-schnell") 
@@ -124,11 +115,12 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
                             
                 with gr.Column():
                     image_out = gr.Image(height=800,sources=[],label="Output", elem_id="output-img",format="jpg")
+                    mask_out = gr.Image(height=800,sources=[],label="Mask", elem_id="mask-img",format="jpeg")
 
                     
             
 
-    btn.click(fn=process_images, inputs=[image, image_mask,prompt,strength,seed,inference_step], outputs =image_out, api_name='infer')
+    #btn.click(fn=process_images, inputs=[image, image_mask,prompt,strength,seed,inference_step], outputs =[image_out,mask_out], api_name='infer')
     gr.Examples(
                examples=[
                     #["images/00547245_99.jpg", "images/00547245_99_mask.jpg","a beautiful girl,eyes closed",0.8,"images/00547245.jpg"],
@@ -146,13 +138,14 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
             
         """
     )
-    """gr.on(
+    gr.on(
         triggers=[btn.click, prompt.submit],
         fn = process_images,
-        inputs = [image, image_mask, prompt, strength, seed],
-        outputs = [image_out]
-    )"""
+        inputs = [image, image_mask,prompt,strength,seed,inference_step],
+        outputs = [image_out,mask_out]
+    )
 
+if __name__ == "__main__":
     demo.launch()