Update block.py

block.py

@@ -1,10 +1,10 @@
-from diffusers.modular_pipelines import PipelineBlock, PipelineState, InputParam, OutputParam
+from diffusers.modular_pipelines import ModularPipelineBlocks, PipelineState, InputParam, OutputParam
 from diffusers.utils import load_image
 from PIL import Image
 import cv2
 import numpy as np
 
-class GetImageStep(PipelineBlock):
+class GetImageStep(ModularPipelineBlocks):
 
     PROCESSOR_IDS = set([
         "canny", "lineart_anime",