Update app.py

app.py

@@ -117,7 +117,7 @@ def extract_middle_slices(nifti_path, output_image_path, slice_size=180, center=
     # Optionally label connected components
     if label_components:
         structure = generate_binary_structure(3, 3)  # 3D, 26-connectivity
-        labeled_data, num_features = label(resampled_data > 0, structure=structure)
+        labeled_data, num_features = label(data > 0, structure=structure)
     else:
         labeled_data = resampled_data
         num_features = None  # Not needed if we're not labeling