requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu118
torch
torchvision==0.18.0
torchaudio==2.3.0

numpy==1.26.4
pyyaml==6.0.1
opencv-python==4.10.0.84
scipy==1.13.1
imageio==2.34.2
lmdb==1.4.1
tqdm==4.66.4
rich==13.7.1
ffmpeg==1.4
onnxruntime-gpu==1.18.0
onnx==1.16.1
scikit-image==0.24.0
albumentations==1.4.10
matplotlib==3.9.0
imageio-ffmpeg==0.5.1
tyro==0.8.5

# ↓ ここを追加 ▼
pydantic==2.8.2
pydantic-core==2.20.1
fastapi==0.112.4
# ↑ ここまで追加

gradio==4.37.1
gdown