Update app.py

app.py

@@ -134,7 +134,7 @@ model = ModelManager(
         device=device,
     )
 
-image_type = 'filepath' #'pil'
+image_type = 'pil' # filepath' #'pil'
 def predict(input):
     if image_type == 'filepath':
         # input: {'image': '/tmp/tmp8mn9xw93.png', 'mask': '/tmp/tmpn5ars4te.png'}
@@ -148,7 +148,8 @@ def predict(input):
         mask_pil = input['mask']
         image = np.array(image_pil)
         mask = np.array(mask_pil.convert("L"))
-    output = model_process(image, mask)
+        output = mask_pil 
+    # output = model_process(image, mask)
     return output
 
 css = '''
@@ -196,7 +197,7 @@ with image_blocks as demo:
         with gr.Box():
             with gr.Row():
                 with gr.Column():
-                    image = gr.Image(source='upload', elem_id="image_upload", shape=(512,512),tool='sketch', type=f'{image_type}', label="Upload").style(mobile_collapse=False, height=512)
+                    image = gr.Image(source='upload', elem_id="image_upload",tool='sketch', type=f'{image_type}', label="Upload").style(mobile_collapse=False, height=512)
                     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
                         btn_in = gr.Button("Done!").style(
                             margin=True,