Update app.py

app.py

@@ -199,7 +199,7 @@ def model_process(input):
     start = time.time()
     res_np_img = model(image, mask, config)
     logger.info(f"process time: {(time.time() - start) * 1000}ms, {res_np_img.shape}")
-    print(f"process time: {(time.time() - start) * 1000}ms, {res_np_img.shape} / {res_np_img[250][250]}")
+    print(f"process time_1_: {(time.time() - start) * 1000}ms, {alpha_channel.shape}, {res_np_img.shape} / {res_np_img[250][250]}")
 
     torch.cuda.empty_cache()
 
@@ -211,6 +211,7 @@ def model_process(input):
         res_np_img = np.concatenate(
             (res_np_img, alpha_channel[:, :, np.newaxis]), axis=-1
         )
+    print(f"process time_2_: {(time.time() - start) * 1000}ms, {res_np_img.shape} / {res_np_img[250][250]}")
     image = Image.fromarray(res_np_img)
     image.save(f'./result_image.png')
     return image