Update app.py

app.py

@@ -216,9 +216,9 @@ 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}")
-    image = Image.fromarray(res_np_img.astype(np.uint8))
+    image = Image.open(io.BytesIO(numpy_to_bytes(res_np_img, ext)))
     image.save(f'./result_image.png')
-    return image_pil # res_np_img.astype(np.uint8) # image
+    return image # res_np_img.astype(np.uint8) # image
     
     '''
     ext = get_image_ext(origin_image_bytes)