wip

lama_cleaner/model/opencv2.py

@@ -1,6 +1,6 @@
 import cv2
 from lama_cleaner.model.base import InpaintModel
-from lama_cleaner.schema import Config
+from lama_cleaner.schema import InpaintRequest
 
 flag_map = {"INPAINT_NS": cv2.INPAINT_NS, "INPAINT_TELEA": cv2.INPAINT_TELEA}
 
@@ -14,7 +14,7 @@ class OpenCV2(InpaintModel):
     def is_downloaded() -> bool:
         return True
 
-    def forward(self, image, mask, config: Config):
+    def forward(self, image, mask, config: InpaintRequest):
         """Input image and output image have same size
         image: [H, W, C] RGB
         mask: [H, W, 1]