Update app.py

app.py

@@ -139,7 +139,7 @@ def model_process(input):
     #image, alpha_channel = load_img(image)
     # Origin image shape: (512, 512, 3)
     
-    alpha_channel = np.ones((image.shape[0],image.shape[1]))*255
+    alpha_channel = (np.ones((image.shape[0],image.shape[1]))*255).astype(np.uint8)
     original_shape = image.shape
     interpolation = cv2.INTER_CUBIC    
     
@@ -218,7 +218,7 @@ def model_process(input):
     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.save(f'./result_image.png')
-    return image
+    return res_np_img.astype(np.uint8) # image
     
     '''
     ext = get_image_ext(origin_image_bytes)