Update app.py

app.py

@@ -8,7 +8,7 @@ import nibabel as nib
 import matplotlib.pyplot as plt
 import spaces  # Import spaces for GPU decoration
 import numpy as np
-from scipy.ndimage import center_of_mass, zoom
+from scipy.ndimage import center_of_mass, zoom, label, generate_binary_structure
 
 # Define paths
 MODEL_DIR = "./model"  # Local directory to store the downloaded model