Update src/gradio_pipeline.py

src/gradio_pipeline.py

@@ -6,6 +6,7 @@ Pipeline for gradio
 import gradio as gr
 from .config.argument_config import ArgumentConfig
 from .live_portrait_pipeline import LivePortraitPipeline
+from .live_portrait_cpu_pipeline import LiveCPUPortraitPipeline
 from .utils.io import load_img_online
 from .utils.rprint import rlog as log
 from .utils.crop import prepare_paste_back, paste_back
@@ -19,7 +20,7 @@ def update_args(args, user_args):
             setattr(args, k, v)
     return args
 
-class GradioPipeline(LivePortraitPipeline):
+class GradioPipeline(LiveCPUPortraitPipeline):
 
     def __init__(self, inference_cfg, crop_cfg, args: ArgumentConfig):
         super().__init__(inference_cfg, crop_cfg)