Update app.py

app.py

@@ -223,7 +223,7 @@ def model_process(input): #image, mask):
     # {'image': '/tmp/tmp8mn9xw93.png', 'mask': '/tmp/tmpn5ars4te.png'}
     # input = request.files
     # RGB
-    origin_image_bytes = input["image"].read()
+    origin_image_bytes = open(input["image"],'rb').read()
     print(f'origin_image_bytes = ', type(origin_image_bytes), len(origin_image_bytes))
 
     image, alpha_channel = load_img(origin_image_bytes)
@@ -274,7 +274,7 @@ def model_process(input): #image, mask):
     logger.info(f"Resized image shape: {image.shape} / {type(image)}")
     print(f"Resized image shape: {image.shape} / {image[250][250]}")
 
-    mask, _ = load_img(input["mask"].read(), gray=True)
+    mask, _ = load_img(open(input["mask"],'rb').read(), gray=True)
     mask = resize_max_size(mask, size_limit=size_limit, interpolation=interpolation)
     print(f"mask image shape: {mask.shape} / {type(mask)} / {mask[250][250]}")
 
@@ -339,8 +339,8 @@ def read_content(file_path: str) -> str:
 
     return content
 
-def predict(dict):
-    print(f'liuyz_0_', dict)  
+def predict(input):
+    print(f'liuyz_0_', input)  
      
     '''
     image = dict["image"] # .convert("RGB") #.resize((512, 512))
@@ -351,7 +351,7 @@ def predict(dict):
     # mask = dict["mask"] # .convert("RGB") #.resize((512, 512))
     '''
    
-    output = model_process(dict) # dict["image"], dict["mask"])
+    output = model_process(input) # dict["image"], dict["mask"])
     # output = mask #output.images[0]
     # output = pipe(prompt = prompt, image=init_image, mask_image=mask,guidance_scale=7.5)
     
@@ -447,7 +447,7 @@ with image_blocks as demo:
         with gr.Box():
             with gr.Row():
                 with gr.Column():
-                    image = gr.Image(source='upload', tool='sketch', type='file',label="Upload").style(height=512)
+                    image = gr.Image(source='upload', tool='sketch', type='filepath',label="Upload").style(height=512)
                     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
                         # prompt = gr.Textbox(placeholder = 'Your prompt (what you want in place of what is erased)', show_label=False, elem_id="input-text")
                         btn = gr.Button("Done!").style(