Spaces:
Runtime error
Runtime error
Commit
·
9074b85
0
Parent(s):
Duplicate from kevinwang676/ChatGLM2-SadTalker
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- .gitattributes +52 -0
- .gitignore +159 -0
- Dockerfile +59 -0
- LICENSE +21 -0
- README.md +15 -0
- app.py +179 -0
- checkpoints/BFM_Fitting/01_MorphableModel.mat +1 -0
- checkpoints/BFM_Fitting/BFM09_model_info.mat +1 -0
- checkpoints/BFM_Fitting/BFM_exp_idx.mat +1 -0
- checkpoints/BFM_Fitting/BFM_front_idx.mat +1 -0
- checkpoints/BFM_Fitting/Exp_Pca.bin +1 -0
- checkpoints/BFM_Fitting/facemodel_info.mat +1 -0
- checkpoints/BFM_Fitting/select_vertex_id.mat +1 -0
- checkpoints/BFM_Fitting/similarity_Lm3D_all.mat +1 -0
- checkpoints/BFM_Fitting/std_exp.txt +1 -0
- checkpoints/auido2exp_00300-model.pth +1 -0
- checkpoints/auido2pose_00140-model.pth +1 -0
- checkpoints/epoch_20.pth +1 -0
- checkpoints/facevid2vid_00189-model.pth.tar +1 -0
- checkpoints/hub/checkpoints/2DFAN4-cd938726ad.zip +1 -0
- checkpoints/hub/checkpoints/s3fd-619a316812.pth +1 -0
- checkpoints/mapping_00229-model.pth.tar +1 -0
- checkpoints/shape_predictor_68_face_landmarks.dat +1 -0
- checkpoints/wav2lip.pth +1 -0
- docs/sadtalker_logo.png +0 -0
- examples/driven_audio/RD_Radio31_000.wav +0 -0
- examples/driven_audio/RD_Radio34_002.wav +0 -0
- examples/driven_audio/RD_Radio36_000.wav +0 -0
- examples/driven_audio/RD_Radio40_000.wav +0 -0
- examples/driven_audio/bus_chinese.wav +0 -0
- examples/driven_audio/chinese_news.wav +3 -0
- examples/driven_audio/chinese_poem1.wav +0 -0
- examples/driven_audio/chinese_poem2.wav +0 -0
- examples/driven_audio/deyu.wav +3 -0
- examples/driven_audio/eluosi.wav +3 -0
- examples/driven_audio/fayu.wav +3 -0
- examples/driven_audio/imagine.wav +3 -0
- examples/driven_audio/itosinger1.wav +0 -0
- examples/driven_audio/japanese.wav +3 -0
- examples/source_image/art_0.png +0 -0
- examples/source_image/art_1.png +0 -0
- examples/source_image/art_10.png +0 -0
- examples/source_image/art_11.png +0 -0
- examples/source_image/art_12.png +0 -0
- examples/source_image/art_13.png +0 -0
- examples/source_image/art_14.png +0 -0
- examples/source_image/art_15.png +0 -0
- examples/source_image/art_16.png +3 -0
- examples/source_image/art_17.png +3 -0
- examples/source_image/art_18.png +0 -0
.gitattributes
ADDED
|
@@ -0,0 +1,52 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
checkpoints/BFM_Fitting/01_MorphableModel.mat filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
checkpoints/BFM_Fitting/BFM09_model_info.mat filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
checkpoints/facevid2vid_00189-model.pth.tar filter=lfs diff=lfs merge=lfs -text
|
| 38 |
+
checkpoints/mapping_00229-model.pth.tar filter=lfs diff=lfs merge=lfs -text
|
| 39 |
+
checkpoints/shape_predictor_68_face_landmarks.dat filter=lfs diff=lfs merge=lfs -text
|
| 40 |
+
examples/driven_audio/chinese_news.wav filter=lfs diff=lfs merge=lfs -text
|
| 41 |
+
examples/driven_audio/deyu.wav filter=lfs diff=lfs merge=lfs -text
|
| 42 |
+
examples/driven_audio/eluosi.wav filter=lfs diff=lfs merge=lfs -text
|
| 43 |
+
examples/driven_audio/fayu.wav filter=lfs diff=lfs merge=lfs -text
|
| 44 |
+
examples/driven_audio/imagine.wav filter=lfs diff=lfs merge=lfs -text
|
| 45 |
+
examples/driven_audio/japanese.wav filter=lfs diff=lfs merge=lfs -text
|
| 46 |
+
examples/source_image/art_16.png filter=lfs diff=lfs merge=lfs -text
|
| 47 |
+
examples/source_image/art_17.png filter=lfs diff=lfs merge=lfs -text
|
| 48 |
+
examples/source_image/art_3.png filter=lfs diff=lfs merge=lfs -text
|
| 49 |
+
examples/source_image/art_4.png filter=lfs diff=lfs merge=lfs -text
|
| 50 |
+
examples/source_image/art_5.png filter=lfs diff=lfs merge=lfs -text
|
| 51 |
+
examples/source_image/art_8.png filter=lfs diff=lfs merge=lfs -text
|
| 52 |
+
examples/source_image/art_9.png filter=lfs diff=lfs merge=lfs -text
|
.gitignore
ADDED
|
@@ -0,0 +1,159 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Byte-compiled / optimized / DLL files
|
| 2 |
+
__pycache__/
|
| 3 |
+
*.py[cod]
|
| 4 |
+
*$py.class
|
| 5 |
+
|
| 6 |
+
# C extensions
|
| 7 |
+
*.so
|
| 8 |
+
|
| 9 |
+
# Distribution / packaging
|
| 10 |
+
.Python
|
| 11 |
+
build/
|
| 12 |
+
develop-eggs/
|
| 13 |
+
dist/
|
| 14 |
+
downloads/
|
| 15 |
+
eggs/
|
| 16 |
+
.eggs/
|
| 17 |
+
lib/
|
| 18 |
+
lib64/
|
| 19 |
+
parts/
|
| 20 |
+
sdist/
|
| 21 |
+
var/
|
| 22 |
+
wheels/
|
| 23 |
+
share/python-wheels/
|
| 24 |
+
*.egg-info/
|
| 25 |
+
.installed.cfg
|
| 26 |
+
*.egg
|
| 27 |
+
MANIFEST
|
| 28 |
+
|
| 29 |
+
# PyInstaller
|
| 30 |
+
# Usually these files are written by a python script from a template
|
| 31 |
+
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
| 32 |
+
*.manifest
|
| 33 |
+
*.spec
|
| 34 |
+
|
| 35 |
+
# Installer logs
|
| 36 |
+
pip-log.txt
|
| 37 |
+
pip-delete-this-directory.txt
|
| 38 |
+
|
| 39 |
+
# Unit test / coverage reports
|
| 40 |
+
htmlcov/
|
| 41 |
+
.tox/
|
| 42 |
+
.nox/
|
| 43 |
+
.coverage
|
| 44 |
+
.coverage.*
|
| 45 |
+
.cache
|
| 46 |
+
nosetests.xml
|
| 47 |
+
coverage.xml
|
| 48 |
+
*.cover
|
| 49 |
+
*.py,cover
|
| 50 |
+
.hypothesis/
|
| 51 |
+
.pytest_cache/
|
| 52 |
+
cover/
|
| 53 |
+
|
| 54 |
+
# Translations
|
| 55 |
+
*.mo
|
| 56 |
+
*.pot
|
| 57 |
+
|
| 58 |
+
# Django stuff:
|
| 59 |
+
*.log
|
| 60 |
+
local_settings.py
|
| 61 |
+
db.sqlite3
|
| 62 |
+
db.sqlite3-journal
|
| 63 |
+
|
| 64 |
+
# Flask stuff:
|
| 65 |
+
instance/
|
| 66 |
+
.webassets-cache
|
| 67 |
+
|
| 68 |
+
# Scrapy stuff:
|
| 69 |
+
.scrapy
|
| 70 |
+
|
| 71 |
+
# Sphinx documentation
|
| 72 |
+
docs/_build/
|
| 73 |
+
|
| 74 |
+
# PyBuilder
|
| 75 |
+
.pybuilder/
|
| 76 |
+
target/
|
| 77 |
+
|
| 78 |
+
# Jupyter Notebook
|
| 79 |
+
.ipynb_checkpoints
|
| 80 |
+
|
| 81 |
+
# IPython
|
| 82 |
+
profile_default/
|
| 83 |
+
ipython_config.py
|
| 84 |
+
|
| 85 |
+
# pyenv
|
| 86 |
+
# For a library or package, you might want to ignore these files since the code is
|
| 87 |
+
# intended to run in multiple environments; otherwise, check them in:
|
| 88 |
+
# .python-version
|
| 89 |
+
|
| 90 |
+
# pipenv
|
| 91 |
+
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
| 92 |
+
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
| 93 |
+
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
| 94 |
+
# install all needed dependencies.
|
| 95 |
+
#Pipfile.lock
|
| 96 |
+
|
| 97 |
+
# poetry
|
| 98 |
+
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
|
| 99 |
+
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
| 100 |
+
# commonly ignored for libraries.
|
| 101 |
+
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
|
| 102 |
+
#poetry.lock
|
| 103 |
+
|
| 104 |
+
# pdm
|
| 105 |
+
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
|
| 106 |
+
#pdm.lock
|
| 107 |
+
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
|
| 108 |
+
# in version control.
|
| 109 |
+
# https://pdm.fming.dev/#use-with-ide
|
| 110 |
+
.pdm.toml
|
| 111 |
+
|
| 112 |
+
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
|
| 113 |
+
__pypackages__/
|
| 114 |
+
|
| 115 |
+
# Celery stuff
|
| 116 |
+
celerybeat-schedule
|
| 117 |
+
celerybeat.pid
|
| 118 |
+
|
| 119 |
+
# SageMath parsed files
|
| 120 |
+
*.sage.py
|
| 121 |
+
|
| 122 |
+
# Environments
|
| 123 |
+
.env
|
| 124 |
+
.venv
|
| 125 |
+
env/
|
| 126 |
+
venv/
|
| 127 |
+
ENV/
|
| 128 |
+
env.bak/
|
| 129 |
+
venv.bak/
|
| 130 |
+
|
| 131 |
+
# Spyder project settings
|
| 132 |
+
.spyderproject
|
| 133 |
+
.spyproject
|
| 134 |
+
|
| 135 |
+
# Rope project settings
|
| 136 |
+
.ropeproject
|
| 137 |
+
|
| 138 |
+
# mkdocs documentation
|
| 139 |
+
/site
|
| 140 |
+
|
| 141 |
+
# mypy
|
| 142 |
+
.mypy_cache/
|
| 143 |
+
.dmypy.json
|
| 144 |
+
dmypy.json
|
| 145 |
+
|
| 146 |
+
# Pyre type checker
|
| 147 |
+
.pyre/
|
| 148 |
+
|
| 149 |
+
# pytype static type analyzer
|
| 150 |
+
.pytype/
|
| 151 |
+
|
| 152 |
+
# Cython debug symbols
|
| 153 |
+
cython_debug/
|
| 154 |
+
|
| 155 |
+
results/
|
| 156 |
+
checkpoints/
|
| 157 |
+
gradio_cached_examples/
|
| 158 |
+
gfpgan/
|
| 159 |
+
start.sh
|
Dockerfile
ADDED
|
@@ -0,0 +1,59 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
|
| 2 |
+
ENV DEBIAN_FRONTEND=noninteractive
|
| 3 |
+
RUN apt-get update && \
|
| 4 |
+
apt-get upgrade -y && \
|
| 5 |
+
apt-get install -y --no-install-recommends \
|
| 6 |
+
git \
|
| 7 |
+
zip \
|
| 8 |
+
unzip \
|
| 9 |
+
git-lfs \
|
| 10 |
+
wget \
|
| 11 |
+
curl \
|
| 12 |
+
# ffmpeg \
|
| 13 |
+
ffmpeg \
|
| 14 |
+
x264 \
|
| 15 |
+
# python build dependencies \
|
| 16 |
+
build-essential \
|
| 17 |
+
libssl-dev \
|
| 18 |
+
zlib1g-dev \
|
| 19 |
+
libbz2-dev \
|
| 20 |
+
libreadline-dev \
|
| 21 |
+
libsqlite3-dev \
|
| 22 |
+
libncursesw5-dev \
|
| 23 |
+
xz-utils \
|
| 24 |
+
tk-dev \
|
| 25 |
+
libxml2-dev \
|
| 26 |
+
libxmlsec1-dev \
|
| 27 |
+
libffi-dev \
|
| 28 |
+
liblzma-dev && \
|
| 29 |
+
apt-get clean && \
|
| 30 |
+
rm -rf /var/lib/apt/lists/*
|
| 31 |
+
|
| 32 |
+
RUN useradd -m -u 1000 user
|
| 33 |
+
USER user
|
| 34 |
+
ENV HOME=/home/user \
|
| 35 |
+
PATH=/home/user/.local/bin:${PATH}
|
| 36 |
+
WORKDIR ${HOME}/app
|
| 37 |
+
|
| 38 |
+
RUN curl https://pyenv.run | bash
|
| 39 |
+
ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
|
| 40 |
+
ENV PYTHON_VERSION=3.10.9
|
| 41 |
+
RUN pyenv install ${PYTHON_VERSION} && \
|
| 42 |
+
pyenv global ${PYTHON_VERSION} && \
|
| 43 |
+
pyenv rehash && \
|
| 44 |
+
pip install --no-cache-dir -U pip setuptools wheel
|
| 45 |
+
|
| 46 |
+
RUN pip install --no-cache-dir -U torch==1.12.1 torchvision==0.13.1
|
| 47 |
+
COPY --chown=1000 requirements.txt /tmp/requirements.txt
|
| 48 |
+
RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
|
| 49 |
+
|
| 50 |
+
COPY --chown=1000 . ${HOME}/app
|
| 51 |
+
RUN ls -a
|
| 52 |
+
ENV PYTHONPATH=${HOME}/app \
|
| 53 |
+
PYTHONUNBUFFERED=1 \
|
| 54 |
+
GRADIO_ALLOW_FLAGGING=never \
|
| 55 |
+
GRADIO_NUM_PORTS=1 \
|
| 56 |
+
GRADIO_SERVER_NAME=0.0.0.0 \
|
| 57 |
+
GRADIO_THEME=huggingface \
|
| 58 |
+
SYSTEM=spaces
|
| 59 |
+
CMD ["python", "app.py"]
|
LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
MIT License
|
| 2 |
+
|
| 3 |
+
Copyright (c) 2023 Tencent AI Lab
|
| 4 |
+
|
| 5 |
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
| 6 |
+
of this software and associated documentation files (the "Software"), to deal
|
| 7 |
+
in the Software without restriction, including without limitation the rights
|
| 8 |
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
| 9 |
+
copies of the Software, and to permit persons to whom the Software is
|
| 10 |
+
furnished to do so, subject to the following conditions:
|
| 11 |
+
|
| 12 |
+
The above copyright notice and this permission notice shall be included in all
|
| 13 |
+
copies or substantial portions of the Software.
|
| 14 |
+
|
| 15 |
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
| 16 |
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
| 17 |
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
| 18 |
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
| 19 |
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
| 20 |
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
| 21 |
+
SOFTWARE.
|
README.md
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
title: SadTalker
|
| 3 |
+
emoji: 😭
|
| 4 |
+
colorFrom: purple
|
| 5 |
+
colorTo: green
|
| 6 |
+
sdk: gradio
|
| 7 |
+
sdk_version: 3.23.0
|
| 8 |
+
app_file: app.py
|
| 9 |
+
pinned: false
|
| 10 |
+
license: mit
|
| 11 |
+
duplicated_from: kevinwang676/ChatGLM2-SadTalker
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
|
| 15 |
+
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
app.py
ADDED
|
@@ -0,0 +1,179 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import os, sys
|
| 2 |
+
import tempfile
|
| 3 |
+
import gradio as gr
|
| 4 |
+
from src.gradio_demo import SadTalker
|
| 5 |
+
# from src.utils.text2speech import TTSTalker
|
| 6 |
+
from huggingface_hub import snapshot_download
|
| 7 |
+
|
| 8 |
+
def get_source_image(image):
|
| 9 |
+
return image
|
| 10 |
+
|
| 11 |
+
try:
|
| 12 |
+
import webui # in webui
|
| 13 |
+
in_webui = True
|
| 14 |
+
except:
|
| 15 |
+
in_webui = False
|
| 16 |
+
|
| 17 |
+
|
| 18 |
+
def toggle_audio_file(choice):
|
| 19 |
+
if choice == False:
|
| 20 |
+
return gr.update(visible=True), gr.update(visible=False)
|
| 21 |
+
else:
|
| 22 |
+
return gr.update(visible=False), gr.update(visible=True)
|
| 23 |
+
|
| 24 |
+
def ref_video_fn(path_of_ref_video):
|
| 25 |
+
if path_of_ref_video is not None:
|
| 26 |
+
return gr.update(value=True)
|
| 27 |
+
else:
|
| 28 |
+
return gr.update(value=False)
|
| 29 |
+
|
| 30 |
+
def download_model():
|
| 31 |
+
REPO_ID = 'vinthony/SadTalker-V002rc'
|
| 32 |
+
snapshot_download(repo_id=REPO_ID, local_dir='./checkpoints', local_dir_use_symlinks=True)
|
| 33 |
+
|
| 34 |
+
def sadtalker_demo():
|
| 35 |
+
|
| 36 |
+
download_model()
|
| 37 |
+
|
| 38 |
+
sad_talker = SadTalker(lazy_load=True)
|
| 39 |
+
# tts_talker = TTSTalker()
|
| 40 |
+
|
| 41 |
+
download_model()
|
| 42 |
+
sad_talker = SadTalker(lazy_load=True)
|
| 43 |
+
|
| 44 |
+
|
| 45 |
+
with gr.Blocks(analytics_enabled=False) as sadtalker_interface:
|
| 46 |
+
gr.Markdown("<div align='center'> <h2> 😭 SadTalker: Learning Realistic 3D Motion Coefficients for Stylized Audio-Driven Single Image Talking Face Animation (CVPR 2023) </span> </h2> \
|
| 47 |
+
<a style='font-size:18px;color: #efefef' href='https://arxiv.org/abs/2211.12194'>Arxiv</a> \
|
| 48 |
+
<a style='font-size:18px;color: #efefef' href='https://sadtalker.github.io'>Homepage</a> \
|
| 49 |
+
<a style='font-size:18px;color: #efefef' href='https://github.com/Winfredy/SadTalker'> Github </div>")
|
| 50 |
+
|
| 51 |
+
|
| 52 |
+
gr.Markdown("""
|
| 53 |
+
<b>You may duplicate the space and upgrade to GPU in settings for better performance and faster inference without waiting in the queue. <a style='display:inline-block' href="https://huggingface.co/spaces/vinthony/SadTalker?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a></b> \
|
| 54 |
+
<br/><b>Alternatively, try our GitHub <a href=https://github.com/Winfredy/SadTalker> code </a> on your own GPU. </b> <a style='display:inline-block' href="https://github.com/Winfredy/SadTalker"><img src="https://img.shields.io/github/stars/Winfredy/SadTalker?style=social"/></a> \
|
| 55 |
+
""")
|
| 56 |
+
|
| 57 |
+
with gr.Row().style(equal_height=False):
|
| 58 |
+
with gr.Column(variant='panel'):
|
| 59 |
+
with gr.Tabs(elem_id="sadtalker_source_image"):
|
| 60 |
+
with gr.TabItem('Source image'):
|
| 61 |
+
with gr.Row():
|
| 62 |
+
source_image = gr.Image(label="Source image", source="upload", type="filepath", elem_id="img2img_image").style(width=512)
|
| 63 |
+
|
| 64 |
+
|
| 65 |
+
with gr.Tabs(elem_id="sadtalker_driven_audio"):
|
| 66 |
+
with gr.TabItem('Driving Methods'):
|
| 67 |
+
|
| 68 |
+
with gr.Row():
|
| 69 |
+
driven_audio = gr.Audio(label="Input audio", source="upload", type="filepath")
|
| 70 |
+
driven_audio_no = gr.Audio(label="Use IDLE mode, no audio is required", source="upload", type="filepath", visible=False)
|
| 71 |
+
|
| 72 |
+
with gr.Column():
|
| 73 |
+
use_idle_mode = gr.Checkbox(label="Use Idle Animation", visible=False)
|
| 74 |
+
length_of_audio = gr.Number(value=5, label="The length(seconds) of the generated video.", visible=False)
|
| 75 |
+
use_idle_mode.change(toggle_audio_file, inputs=use_idle_mode, outputs=[driven_audio, driven_audio_no]) # todo
|
| 76 |
+
|
| 77 |
+
with gr.Row():
|
| 78 |
+
ref_video = gr.Video(label="Reference Video", source="upload", type="filepath", elem_id="vidref", visible=False).style(width=512)
|
| 79 |
+
|
| 80 |
+
with gr.Column():
|
| 81 |
+
use_ref_video = gr.Checkbox(label="Use Reference Video", visible=False)
|
| 82 |
+
ref_info = gr.Radio(['pose', 'blink','pose+blink', 'all'], value='pose', label='Reference Video',info="How to borrow from reference Video?((fully transfer, aka, video driving mode))", visible=False)
|
| 83 |
+
|
| 84 |
+
ref_video.change(ref_video_fn, inputs=ref_video, outputs=[use_ref_video]) # todo
|
| 85 |
+
|
| 86 |
+
|
| 87 |
+
with gr.Column(variant='panel'):
|
| 88 |
+
with gr.Tabs(elem_id="sadtalker_checkbox"):
|
| 89 |
+
with gr.TabItem('Settings'):
|
| 90 |
+
with gr.Column(variant='panel'):
|
| 91 |
+
# width = gr.Slider(minimum=64, elem_id="img2img_width", maximum=2048, step=8, label="Manually Crop Width", value=512) # img2img_width
|
| 92 |
+
# height = gr.Slider(minimum=64, elem_id="img2img_height", maximum=2048, step=8, label="Manually Crop Height", value=512) # img2img_width
|
| 93 |
+
with gr.Row():
|
| 94 |
+
pose_style = gr.Slider(minimum=0, maximum=45, step=1, label="Pose style", value=0, visible=False) #
|
| 95 |
+
exp_weight = gr.Slider(minimum=0, maximum=3, step=0.1, label="expression scale", value=1, visible=False) #
|
| 96 |
+
blink_every = gr.Checkbox(label="use eye blink", value=True)
|
| 97 |
+
|
| 98 |
+
with gr.Row():
|
| 99 |
+
size_of_image = gr.Radio([256, 512], value=256, label='face model resolution', info="use 256/512 model?", visible=False) #
|
| 100 |
+
preprocess_type = gr.Radio(['crop', 'resize','full', 'extcrop', 'extfull'], value='crop', label='preprocess', info="How to handle input image?", visible=False)
|
| 101 |
+
|
| 102 |
+
with gr.Row():
|
| 103 |
+
is_still_mode = gr.Checkbox(label="Still Mode (fewer head motion, works with preprocess `full`)", value=True, visible=False)
|
| 104 |
+
facerender = gr.Radio(['facevid2vid','pirender'], value='facevid2vid', label='facerender', info="which face render?", visible=False)
|
| 105 |
+
|
| 106 |
+
with gr.Row():
|
| 107 |
+
batch_size = gr.Slider(label="batch size in generation", step=1, maximum=10, value=1)
|
| 108 |
+
enhancer = gr.Checkbox(label="GFPGAN as Face enhancer", value=True, visible=False)
|
| 109 |
+
|
| 110 |
+
submit = gr.Button('Generate', elem_id="sadtalker_generate", variant='primary')
|
| 111 |
+
|
| 112 |
+
with gr.Tabs(elem_id="sadtalker_genearted"):
|
| 113 |
+
gen_video = gr.Video(label="Generated video", format="mp4").style(width=256)
|
| 114 |
+
|
| 115 |
+
|
| 116 |
+
|
| 117 |
+
submit.click(
|
| 118 |
+
fn=sad_talker.test,
|
| 119 |
+
inputs=[source_image,
|
| 120 |
+
driven_audio,
|
| 121 |
+
preprocess_type,
|
| 122 |
+
is_still_mode,
|
| 123 |
+
enhancer,
|
| 124 |
+
batch_size,
|
| 125 |
+
size_of_image,
|
| 126 |
+
pose_style,
|
| 127 |
+
facerender,
|
| 128 |
+
exp_weight,
|
| 129 |
+
use_ref_video,
|
| 130 |
+
ref_video,
|
| 131 |
+
ref_info,
|
| 132 |
+
use_idle_mode,
|
| 133 |
+
length_of_audio,
|
| 134 |
+
blink_every
|
| 135 |
+
],
|
| 136 |
+
outputs=[gen_video]
|
| 137 |
+
)
|
| 138 |
+
|
| 139 |
+
with gr.Row():
|
| 140 |
+
examples = [
|
| 141 |
+
[
|
| 142 |
+
'examples/source_image/full_body_1.png',
|
| 143 |
+
'examples/driven_audio/bus_chinese.wav',
|
| 144 |
+
'crop',
|
| 145 |
+
True,
|
| 146 |
+
False
|
| 147 |
+
],
|
| 148 |
+
[
|
| 149 |
+
'examples/source_image/full_body_2.png',
|
| 150 |
+
'examples/driven_audio/japanese.wav',
|
| 151 |
+
'crop',
|
| 152 |
+
False,
|
| 153 |
+
False
|
| 154 |
+
],
|
| 155 |
+
[
|
| 156 |
+
'examples/source_image/full3.png',
|
| 157 |
+
'examples/driven_audio/deyu.wav',
|
| 158 |
+
'crop',
|
| 159 |
+
False,
|
| 160 |
+
True
|
| 161 |
+
],
|
| 162 |
+
|
| 163 |
+
]
|
| 164 |
+
gr.Examples(examples=examples,
|
| 165 |
+
inputs=[
|
| 166 |
+
source_image,
|
| 167 |
+
driven_audio,
|
| 168 |
+
preprocess_type,
|
| 169 |
+
is_still_mode,
|
| 170 |
+
enhancer],
|
| 171 |
+
outputs=[gen_video],
|
| 172 |
+
fn=sad_talker.test,
|
| 173 |
+
cache_examples=os.getenv('SYSTEM') == 'spaces') #
|
| 174 |
+
|
| 175 |
+
|
| 176 |
+
|
| 177 |
+
sadtalker_interface.queue(max_size=10).launch(debug=True)
|
| 178 |
+
|
| 179 |
+
|
checkpoints/BFM_Fitting/01_MorphableModel.mat
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/37b1f0742db356a3b1568a8365a06f5b0fe0ab687ac1c3068c803666cbd4d8e2
|
checkpoints/BFM_Fitting/BFM09_model_info.mat
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/db8d00544f0b0182f1b8430a3bb87662b3ff674eb33c84e6f52dbe2971adb81b
|
checkpoints/BFM_Fitting/BFM_exp_idx.mat
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/1146e4e9c3bef303a497383aa7974c014fe945c7
|
checkpoints/BFM_Fitting/BFM_front_idx.mat
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/b9d7b0953dd1dc5b1e28144610485409ac321f9b
|
checkpoints/BFM_Fitting/Exp_Pca.bin
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/e7f31380e6cbdaf2aeec698db220bac4f221946e4d551d88c092d47ec49b1726
|
checkpoints/BFM_Fitting/facemodel_info.mat
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/3e516ec7297fa3248098f49ecea10579f4831c0a
|
checkpoints/BFM_Fitting/select_vertex_id.mat
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/5b8b220093d93b133acc94ffed159f31a74854cd
|
checkpoints/BFM_Fitting/similarity_Lm3D_all.mat
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/a0e23588302bc71fc899eef53ff06df5f4df4c1d
|
checkpoints/BFM_Fitting/std_exp.txt
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/767b8de4ea1ca78b6f22b98ff2dee4fa345500bb
|
checkpoints/auido2exp_00300-model.pth
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/b7608f0e6b477e50e03ca569ac5b04a841b9217f89d502862fc78fda4e46dec4
|
checkpoints/auido2pose_00140-model.pth
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/4fba6701852dc57efbed25b1e4276e4ff752941860d69fc4429f08a02326ebce
|
checkpoints/epoch_20.pth
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/6d17a6b23457b521801baae583cb6a58f7238fe6721fc3d65d76407460e9149b
|
checkpoints/facevid2vid_00189-model.pth.tar
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/fbad01d46f0510276dc4521322dde6824a873a4222cd0740c85762e7067ea71d
|
checkpoints/hub/checkpoints/2DFAN4-cd938726ad.zip
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/cd938726adb1f15f361263cce2db9cb820c42585fa8796ec72ce19107f369a46
|
checkpoints/hub/checkpoints/s3fd-619a316812.pth
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/619a31681264d3f7f7fc7a16a42cbbe8b23f31a256f75a366e5a1bcd59b33543
|
checkpoints/mapping_00229-model.pth.tar
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker-V002rc/blobs/62a1e06006cc963220f6477438518ed86e9788226c62ae382ddc42fbcefb83f1
|
checkpoints/shape_predictor_68_face_landmarks.dat
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/fbdc2cb80eb9aa7a758672cbfdda32ba6300efe9b6e6c7a299ff7e736b11b92f
|
checkpoints/wav2lip.pth
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
../../../../root/.cache/huggingface/hub/models--vinthony--SadTalker/blobs/b78b681b68ad9fe6c6fb1debc6ff43ad05834a8af8a62ffc4167b7b34ef63c37
|
docs/sadtalker_logo.png
ADDED
|
examples/driven_audio/RD_Radio31_000.wav
ADDED
|
Binary file (512 kB). View file
|
|
|
examples/driven_audio/RD_Radio34_002.wav
ADDED
|
Binary file (512 kB). View file
|
|
|
examples/driven_audio/RD_Radio36_000.wav
ADDED
|
Binary file (512 kB). View file
|
|
|
examples/driven_audio/RD_Radio40_000.wav
ADDED
|
Binary file (512 kB). View file
|
|
|
examples/driven_audio/bus_chinese.wav
ADDED
|
Binary file (652 kB). View file
|
|
|
examples/driven_audio/chinese_news.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7b0f4d313a1ca671bc4831d60bcf0c12225efbffe6c0e93e54fbfe9bcd4021cb
|
| 3 |
+
size 1536078
|
examples/driven_audio/chinese_poem1.wav
ADDED
|
Binary file (263 kB). View file
|
|
|
examples/driven_audio/chinese_poem2.wav
ADDED
|
Binary file (461 kB). View file
|
|
|
examples/driven_audio/deyu.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ba1839c57770a2ab0b593ce814344bfd4d750da02acc9be9e8cf5b9113a0f88a
|
| 3 |
+
size 2694784
|
examples/driven_audio/eluosi.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d4a3593815dc7b68c256672baa61934c9479efa770af2065fb0886f02713606e
|
| 3 |
+
size 1786672
|
examples/driven_audio/fayu.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:16ebd13626ae4171030b4ea05cceef06078483c352e4b68d469fc2a52bfffceb
|
| 3 |
+
size 1940428
|
examples/driven_audio/imagine.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2db410217e074d91ae6011e1c5dc0b94f02d05d381c50af8e54253eeacad17d2
|
| 3 |
+
size 1618510
|
examples/driven_audio/itosinger1.wav
ADDED
|
Binary file (500 kB). View file
|
|
|
examples/driven_audio/japanese.wav
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3db5426d0b158799e2be4f609b11f75bfbd4affffe18e9a1c8e6f241fcdedcfc
|
| 3 |
+
size 2622712
|
examples/source_image/art_0.png
ADDED
|
examples/source_image/art_1.png
ADDED
|
examples/source_image/art_10.png
ADDED
|
examples/source_image/art_11.png
ADDED
|
examples/source_image/art_12.png
ADDED
|
examples/source_image/art_13.png
ADDED
|
examples/source_image/art_14.png
ADDED
|
examples/source_image/art_15.png
ADDED
|
examples/source_image/art_16.png
ADDED
|
Git LFS Details
|
examples/source_image/art_17.png
ADDED
|
Git LFS Details
|
examples/source_image/art_18.png
ADDED
|