Spaces:

Deevyankar
/

Deep-AD

Running

Deevyankar commited on Feb 5

Commit

d23d6c6

verified ·

1 Parent(s): cc95a98

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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 = processed_image.to_nibabel()
     saved_preprocessed_path = 'input_image'
     nib.save(a, saved_preprocessed_path)

     element2 = st.pyplot(ants.plot(display_image))
     processed_image = pre_process(saved_path)
+    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)