Update app.py

app.py

@@ -124,7 +124,7 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180, center=
 
     # Compute or reuse the center of mass
     if center is None:
-        com = center_of_mass(labeled_data > 0)
+        com = center_of_mass(resampled_data > 0)
         center = np.round(com).astype(int)
 
     # Define half the slice size
@@ -151,9 +151,9 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180, center=
         return padded_slice
 
     # Extract slices
-    axial_slice = extract_2d_slice(labeled_data, center, axis=2)
-    coronal_slice = extract_2d_slice(labeled_data, center, axis=1)
-    sagittal_slice = extract_2d_slice(labeled_data, center, axis=0)
+    axial_slice = extract_2d_slice(resampled_data, center, axis=2)
+    coronal_slice = extract_2d_slice(resampled_data, center, axis=1)
+    sagittal_slice = extract_2d_slice(resampled_data, center, axis=0)
 
     # Apply rotations
     axial_slice = np.rot90(axial_slice, k=-1)