Update app.py

app.py

@@ -14,7 +14,6 @@ import antspynet as antspynet
 from torch.utils.data import DataLoader
 from monai.transforms import Compose, LoadImaged
 from monai.networks.nets.efficientnet import EfficientNetBN
-import nibabel as nib
 import dicom2nifti
 
 
@@ -108,7 +107,8 @@ if len(file_upload) == 1:
     element2 = st.pyplot(ants.plot(display_image))
 
     processed_image = pre_process(saved_path)
-    a = nib.load(f"{processed_image}")#processed_image.to_nibabel()
+    a=ants.utils.convert_nibabel.to_nibabel(processed_image)
+    #a = processed_image.to_nibabel()
     saved_preprocessed_path = 'input_image'
     nib.save(a, saved_preprocessed_path)
     element3 = st.text("Preprocessed Image")