Update block.py
Browse files
block.py
CHANGED
@@ -1,4 +1,7 @@
|
|
1 |
from diffusers.modular_pipelines import PipelineBlock, PipelineState, InputParam, OutputParam
|
|
|
|
|
|
|
2 |
class GetImageStep(PipelineBlock):
|
3 |
|
4 |
PROCESSOR_IDS = set([
|
|
|
1 |
from diffusers.modular_pipelines import PipelineBlock, PipelineState, InputParam, OutputParam
|
2 |
+
from PIL import Image
|
3 |
+
import cv2
|
4 |
+
|
5 |
class GetImageStep(PipelineBlock):
|
6 |
|
7 |
PROCESSOR_IDS = set([
|