Update main.py

main.py

@@ -30,13 +30,12 @@ async def segment(files: list[UploadFile]):
         # with tempfile.NamedTemporaryFile(suffix=".nii") as tempFile:
         #     nib.save(output_img, tempFile.name)
         #     return FileResponse(tempFile.name, media_type="multipart/form-data")
-        
-        output_filename = "output.nii"
-        nib.save(output_img, output_filename)
-        
-        # Return the saved file as a FileResponse
-        return FileResponse(output_filename, media_type="application/octet-stream")
-        
+
+        with tempfile.NamedTemporaryFile(suffix=".nii", delete=False) as tempFile:
+            nib.save(output_img, tempFile.name)
+            tempFile.seek(0)  # Move cursor to the beginning of the file
+            file_content = tempFile.read()
+            return StreamingResponse(content=file_content, media_type="application/octet-stream")
          
     except Exception as e:
         print(f"Error occurred: {str(e)}")