Update app_recon.py

app_recon.py

@@ -172,12 +172,12 @@ def reconstruction(image, files):
     y_max = int(y_nonzero[0].max())
     masked_image, mask = sam_segment(sam_predictor, img.convert('RGB'), x_min, y_min, x_max, y_max)
 
-    mask = mask.resize((width, height), Image.LANCZOS)
+    mask = mask[-1].resize((width, height), Image.LANCZOS)
     depth_np = np.load(files[0])
     normal_np = np.load(files[1])
 
     dir_name = os.path.dirname(os.path.realpath(files[0]))
-    mask_output_temp = mask[-1]
+    mask_output_temp = mask
     name_base = os.path.splitext(os.path.basename(files[0]))[0][:-6]
 
     normal_np[:, :, 0] *= -1