Update main.py

main.py

@@ -36,12 +36,16 @@ def segment(files: list[UploadFile]):
         #     response = FileResponse(tempFile.name, media_type="multipart/form-data")
         #     tempFile.close()
         
-        with tempfile.NamedTemporaryFile(suffix=".nii.gz", delete=False) as tempGzipFile:
-            with gzip.open(tempGzipFile.name, 'wb') as gz:
-                nib.save(output_img, gz)
-            response = FileResponse(tempGzipFile.name, media_type="multipart/form-data")
-            tempGzipFile.close()
-            
+        with tempfile.NamedTemporaryFile(suffix=".nii", delete=False) as tempFile:
+            nib.save(output_img, tempFile.name)
+            tempFile.seek(0)
+            with tempfile.NamedTemporaryFile(suffix=".nii.gz", delete=False) as tempGzipFile:
+                with gzip.open(tempGzipFile.name, 'wb') as gz:
+                    gz.write(tempFile.read())
+                response = FileResponse(tempGzipFile.name, media_type="multipart/form-data")
+                tempGzipFile.close()
+            tempFile.close()    
+                
         is_ready = True
         return response