Update app.py

app.py

@@ -14,7 +14,7 @@ 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 +108,7 @@ if len(file_upload) == 1:
     element2 = st.pyplot(ants.plot(display_image))
 
     processed_image = pre_process(saved_path)
-    a = processed_image.to_nibabel()
+    a = nib.load(processed_image)#processed_image.to_nibabel()
     saved_preprocessed_path = 'input_image'
     nib.save(a, saved_preprocessed_path)
     element3 = st.text("Preprocessed Image")