Update app.py

app.py

@@ -107,7 +107,7 @@ if len(file_upload) == 1:
     element2 = st.pyplot(ants.plot(display_image))
 
     processed_image = pre_process(saved_path)
-    a=nib.Nifti1Image(processed_image)
+    a=ants.image_io.convert_ants_image_to_nibabel(processed_image)
     #a = processed_image.to_nibabel()
     saved_preprocessed_path = 'input_image'
     nib.save(a, saved_preprocessed_path)