READMEに`preload_from_hub`の設定を追加し、`requirements.txt`の依存関係をバージョン指定なしに変更し、互換性のための追加依存関係を追加

README_hf_space.md

@@ -9,6 +9,8 @@ app_file: app.py
 pinned: false
 license: apache-2.0
 hardware: a100-large
+preload_from_hub:
+  - digital-avatar/ditto-talkinghead
 ---
 
 # DittoTalkingHead - Talking Head Generation

packages.txt

@@ -0,0 +1,2 @@
+ffmpeg
+libsndfile1

pre-requirements.txt

@@ -0,0 +1,5 @@
+# Install PyTorch with CUDA support first
+--extra-index-url https://download.pytorch.org/whl/cu121
+torch==2.2.0
+torchvision==0.17.0
+torchaudio==2.2.0

requirements.txt

@@ -1,46 +1,37 @@
 # Core dependencies
-torch==2.5.1
-torchvision==0.20.1
-torchaudio==2.5.1
-numpy==2.0.1
-pillow==11.0.0
+torch
+torchvision
+torchaudio
+numpy
+pillow
 
 # Audio processing
-librosa==0.10.2.post1
-soundfile==0.13.0
-audioread==3.0.1
-soxr==0.5.0.post1
+librosa
+soundfile
+audioread
+soxr
 
 # Video/Image processing
-opencv-python-headless==4.10.0.84
-imageio==2.36.1
-imageio-ffmpeg==0.5.1
-scikit-image==0.25.0
+opencv-python-headless
+imageio
+imageio-ffmpeg
+scikit-image
 
 # Machine learning
-scikit-learn==1.6.0
-scipy==1.15.0
-numba==0.60.0
-
-# TensorRT (GPU acceleration)
-tensorrt==8.6.1
-tensorrt-bindings==8.6.1
-tensorrt-libs==8.6.1
-polygraphy
-colored
+scikit-learn
+scipy
+numba
 
 # Web interface
-gradio==4.19.0
+gradio
 
 # Utilities
-tqdm==4.67.1
-requests==2.32.3
-pyyaml==6.0.2
-joblib==1.4.2
-cython==3.0.11
+tqdm
+requests
+pyyaml
+joblib
+cython
 
-# CUDA dependencies
-cuda-python==12.6.2.post1
-nvidia-cublas-cu12==12.6.4.1
-nvidia-cuda-runtime-cu12==12.6.77
-nvidia-cudnn-cu12==9.6.0.74
+# Additional dependencies for compatibility
+packaging
+typing-extensions