Update inference.py

inference.py

@@ -5,7 +5,7 @@ from src.config.argument_config import ArgumentConfig
 from src.config.inference_config import InferenceConfig
 from src.config.crop_config import CropConfig
 from src.live_portrait_pipeline import LivePortraitPipeline
-
+from src.live_portrait_cpu_pipeline import LiveCPUPortraitPipeline
 
 def partial_fields(target_class, kwargs):
     return target_class(**{k: v for k, v in kwargs.items() if hasattr(target_class, k)})
@@ -20,7 +20,8 @@ def main():
     inference_cfg = partial_fields(InferenceConfig, args.__dict__)  # use attribute of args to initial InferenceConfig
     crop_cfg = partial_fields(CropConfig, args.__dict__)  # use attribute of args to initial CropConfig
 
-    live_portrait_pipeline = LivePortraitPipeline(
+    # live_portrait_pipeline = LivePortraitPipeline(
+    live_portrait_pipeline = LiveCPUPortraitPipeline(
         inference_cfg=inference_cfg,
         crop_cfg=crop_cfg
     )