mirror of
https://github.com/KwaiVGI/LivePortrait.git
synced 2024-12-22 12:22:38 +00:00
fix: clean up
This commit is contained in:
parent
0b303fc39c
commit
eec7aa3337
1
cog.yaml
1
cog.yaml
@ -29,4 +29,5 @@ build:
|
||||
- "matplotlib==3.9.0"
|
||||
- "imageio-ffmpeg==0.5.1"
|
||||
- "tyro==0.8.5"
|
||||
- "gradio==3.48.0"
|
||||
predict: "predict.py:Predictor"
|
||||
|
17
predict.py
17
predict.py
@ -1,9 +1,10 @@
|
||||
from cog import BasePredictor, Input, Path
|
||||
from cog import BasePredictor, Input, Path, File
|
||||
|
||||
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
|
||||
import requests
|
||||
|
||||
|
||||
class Predictor(BasePredictor):
|
||||
@ -16,14 +17,18 @@ class Predictor(BasePredictor):
|
||||
|
||||
def predict(
|
||||
self,
|
||||
image: Path = Input(description="Portrait image"),
|
||||
driving_info: Path = Input(
|
||||
description="driving video or template (.pkl format)"
|
||||
),
|
||||
image: Path = Input(description="Portrait image")
|
||||
) -> Path:
|
||||
"""Run a single prediction on the model"""
|
||||
|
||||
video_path, _ = self.live_portrait_pipeline.execute(
|
||||
ArgumentConfig(source_image=image, driving_info=driving_info, output_dir="/tmp/")
|
||||
ArgumentConfig(
|
||||
source_image=image,
|
||||
driving_info="assets/examples/driving/d0.mp4",
|
||||
output_dir="/tmp/",
|
||||
flag_pasteback=False,
|
||||
flag_do_crop=False,
|
||||
)
|
||||
)
|
||||
|
||||
return Path(video_path)
|
||||
|
Loading…
Reference in New Issue
Block a user