Update app.py

app.py

@@ -215,7 +215,7 @@ def model_process(input):
             (res_np_img, alpha_channel[:, :, np.newaxis]), axis=-1
         )
     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)
+    image = Image.fromarray(res_np_img.astype(np.uint8))
     image.save(f'./result_image.png')
     return image