Update app.py

app.py

@@ -216,6 +216,7 @@ def model_process_pil(input):
         )
         print(f"liuyz_here_40_: {alpha_channel.shape} / {res_np_img.shape} / {alpha_channel.dtype} / {res_np_img.dtype}")
     print(f"process time_2_: {(time.time() - start) * 1000}ms, {alpha_channel.shape}, {res_np_img.shape} / {res_np_img[250][250]} / {res_np_img.dtype}")
+    ext = 'png'
     image = Image.open(io.BytesIO(numpy_to_bytes(res_np_img, ext)))
     image.save(f'./result_image.png')
     return image # res_np_img.astype(np.uint8) # image
@@ -359,7 +360,7 @@ def read_content(file_path):
 
     return content
 
-image_type = 'filepath' #'pil'
+image_type = 'pil'  #'filepath' #'pil'
 def predict(input):
     print(f'liuyz_0_', input)  
     '''