import vtk in prediction.py

Author: baptistebaquero

py/prediction.py

@@ -12,6 +12,7 @@
 from monai.transforms import AsDiscrete
 from monai.data import decollate_batch
 import shutil
+import vtk
 from scipy import linalg
 
 def main(args):