Update app.py

app.py

@@ -8,6 +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
 
 # Define paths
 MODEL_DIR = "./model"  # Local directory to store the downloaded model