[FIX] dockerfile

Dockerfile

@@ -29,25 +29,54 @@ RUN apt-get update && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
+# RUN useradd -m -u 1000 user
+# USER user
+# ENV HOME=/home/user \
+#     PATH=/home/user/.local/bin:${PATH}
+# WORKDIR ${HOME}/app
+
+# RUN curl https://pyenv.run | bash
+# ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
+# ENV PYTHON_VERSION=3.10.18
+# RUN pyenv install ${PYTHON_VERSION} && \
+#     pyenv global ${PYTHON_VERSION} && \
+#     pyenv rehash && \
+#     pip install --no-cache-dir -U pip setuptools wheel
+
+# RUN pip install --user --no-cache-dir -U torch==2.4.0+cu124 torchaudio==2.4.0+cu124 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu124
+# RUN pip install --user --no-cache-dir -U f5_tts==1.1.9
+# COPY --chown=1000 requirements_f5tts.txt /tmp/requirements_f5tts.txt
+# RUN pip install --user --no-cache-dir -U -r /tmp/requirements_f5tts.txt
+# COPY --chown=1000 requirements.txt /tmp/requirements.txt
+# RUN pip install --user --no-cache-dir -U -r /tmp/requirements.txt
+
+# COPY --chown=1000 . ${HOME}/app
+# RUN ls -a
+# ENV PYTHONPATH=${HOME}/app \
+#     PYTHONUNBUFFERED=1 \
+#     GRADIO_ALLOW_FLAGGING=never \
+#     GRADIO_NUM_PORTS=1 \
+#     GRADIO_SERVER_NAME=0.0.0.0 \
+#     GRADIO_THEME=huggingface \
+#     SYSTEM=spaces
+# CMD ["python", "app.py"]
+
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:${PATH}
+    PATH=/home/user/.local/bin:${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
 WORKDIR ${HOME}/app
 
 RUN curl https://pyenv.run | bash
-ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
 ENV PYTHON_VERSION=3.10.18
 RUN pyenv install ${PYTHON_VERSION} && \
     pyenv global ${PYTHON_VERSION} && \
     pyenv rehash && \
-    pip install --no-cache-dir -U pip setuptools wheel
-
-RUN pip install --no-cache-dir -U torch==2.4.0+cu124 torchaudio==2.4.0+cu124 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu124
-COPY --chown=1000 requirements_f5tts.txt /tmp/requirements_f5tts.txt
-RUN pip install --no-cache-dir -U -r /tmp/requirements_f5tts.txt
-COPY --chown=1000 requirements.txt /tmp/requirements.txt
-RUN pip install --no-cache-dir -U -r /tmp/requirements.txt
+    pyenv exec pip install --no-cache-dir -U pip setuptools wheel && \
+    pyenv exec pip install --no-cache-dir -U torch==2.4.0+cu124 torchaudio==2.4.0+cu124 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu124 && \
+    pyenv exec pip install --no-cache-dir -U f5_tts==1.1.9 && \
+    # pyenv exec pip install --no-cache-dir -U -r /tmp/requirements_f5tts.txt && \
+    pyenv exec pip install --no-cache-dir -U -r /tmp/requirements.txt
 
 COPY --chown=1000 . ${HOME}/app
 RUN ls -a

app_tts.py

@@ -8,8 +8,8 @@ from cached_path import cached_path
 import tempfile
 from vinorm import TTSnorm
 from importlib.resources import files
-import sys
-sys.path.append(os.path.join(os.path.dirname(__file__), 'src'))
+# import sys
+# sys.path.append(os.path.join(os.path.dirname(__file__), 'src'))
 from f5_tts.model import DiT
 from f5_tts.infer.utils_infer import (
     preprocess_ref_audio_text,

requirements.txt

@@ -1,71 +1,48 @@
-# CORE compatibility, bắt buộc
-# torch>=2.0.0,<2.5.0
-# torchaudio>=2.0.0,<2.5.0
-# torchvision>=0.15.0,<0.20.0
-f5-tts==1.1.9
-
-# API/Server compatibility (phù hợp với f5-tts)
-fastapi==0.116.1
-starlette==0.47.3
-pydantic<=2.10.6
-
-# Các package TTS/SadTalker, support model
-transformers==4.35.0
-tqdm>=4.65.0
-
-# Các thư viện xử lý ảnh, âm thanh
-imageio
-imageio-ffmpeg
-librosa
-numba
-pydub
-soundfile
-opencv-python
-av
-
-# Các package face/talk/animation
-face_alignment
-basicsr
+# torch==2.4.0
+# torchaudio==2.4.0
+# torchvision==0.19.0
+
+face_alignment==1.3.0
+imageio==2.19.3
+imageio-ffmpeg==0.4.7
+librosa==0.8.0
+numba==0.56.4
+resampy==0.3.1
+pydub==0.25.1 
+scipy
+kornia==0.6.8
+tqdm
+yacs==0.1.8
+pyyaml
+joblib==1.1.0
+scikit-image==0.19.3
+basicsr==1.4.2
 facexlib
 dlib-bin
 gfpgan
-kornia
-scikit-image
-scikit-learn
-yacs
-vocos
-vinorm
-x-transformers>=1.31.14
-
-# Các package phụ trợ (joblib, pandas, ...)
-joblib
-omegaconf
-matplotlib
-pyyaml
-pypinyin
-jieba
-hydra-core
-pyarrow
-soxr
-unidecode
+av
 
-# Đẩy lên cuối, phần nào không critical thì bỏ version, hoặc <= nếu cần fix bug
-huggingface-hub
+safetensors
+gradio
+soundfile
+transformers
+bitsandbytes>0.37.0
+vinorm
+cached_path
+huggingface_hub
 accelerate>=0.33.0
+click
 datasets
-bitsandbytes
+ema_pytorch>=0.5.2
+hydra-core>=1.3.0
+jieba
+matplotlib
+pypinyin
+tomli
+torchdiffeq
+transformers_stream_generator
+vocos
 wandb
+x_transformers>=1.31.14
 
-# Gradio client/UI, DÙNG gradio>=5.0.0 để không conflict f5-tts
-gradio>=5.0.0
-gradio_client
-
-# Dọn dependency system
-protobuf
-filelock
-click
-rich
-requests
-typing_extensions
-python-dateutil
-packaging
+spaces