init

app.py

@@ -237,7 +237,7 @@ class Drag:
         frames = frames.flatten(0, 1)  # [b*13, 3, 256, 256]
         sparse_optical_flow = sparse_optical_flow.flatten(0, 1)  # [b*13, 2, 256, 256]
         mask = mask.flatten(0, 1)  # [b*13, 2, 256, 256]
-        cmp_flow = cmp.run(frames, sparse_optical_flow, mask)  # [b*13, 2, 256, 256]
+        cmp_flow = self.cmp.run(frames, sparse_optical_flow, mask)  # [b*13, 2, 256, 256]
 
         if brush_mask is not None:
             brush_mask = torch.from_numpy(brush_mask) / 255.
@@ -323,7 +323,7 @@ class Drag:
 
         controlnet_flow = torch.where(inmask_no_zero, flow_inmask, flow_outmask)
 
-        val_output = pipeline(
+        val_output = self.pipeline(
             input_first_frame_pil, 
             input_first_frame_pil,
             controlnet_flow,