update

app.py

@@ -23,26 +23,11 @@ def sanitize_prompt(prompt):
   sanitized_prompt = allowed_chars.sub("", prompt)
   return sanitized_prompt
 
-#@spaces.GPU
-def process_img2img(image,mask_image,prompt="a person",strength=0.75,seed=0,num_inference_steps=4):
-    print("start process image process_image")
-    if image == None:
-        print("empty input image returned")
-        return None
-
-    generators = []
-    generator = torch.Generator(device).manual_seed(seed)
-    generators.append(generator)
-    # 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=256)
-
-    # TODO support mask
-    return output.images[0]
-
-
-@spaces.GPU(duration=180)
+
+
+
+
+@spaces.GPU(duration=160)
 def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
     print("start process_images")
     
@@ -65,6 +50,23 @@ def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,progr
         mask = image['layers'][0]
 
 
+    def process_img2img(image,mask_image,prompt="a person",strength=0.75,seed=0,num_inference_steps=4):
+        print("start  process_img2img")
+        if image == None:
+            print("empty input image returned")
+            return None
+
+        generators = []
+        generator = torch.Generator(device).manual_seed(seed)
+        generators.append(generator)
+        # 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)
+
+        # TODO support mask
+        return output.images[0]
+
     output = process_img2img(image["background"],mask,prompt,strength,seed)
    
     print("end process_images")
@@ -125,7 +127,7 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
                     
             
 
-    #btn.click(fn=process_images, inputs=[image, image_mask,prompt,strength,seed], outputs =image_out, api_name='infer')
+    btn.click(fn=process_images, inputs=[image, image_mask,prompt,strength,seed], outputs =image_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"],
@@ -143,12 +145,12 @@ 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]
-    )
+    )"""
 
     demo.launch()