image-seg-inpaint-demo / requirements.txt
coconutiscoding's picture
change model path
5809e83
raw
history blame contribute delete
434 Bytes
# ---- PyTorch (CPU 版本,适配 Hugging Face Space 免费环境) ----
torch==2.0.1+cpu
torchvision==0.15.2+cpu
-f https://download.pytorch.org/whl/torch_stable.html
numpy==1.24.4
# ---- Transformers (分割模型用) ----
transformers
timm # DETR 依赖
# ---- 图像处理 ----
Pillow
opencv-python
# ---- Web 界面 ----
gradio>=4.0.0
# ---- LaMa inpainting 后续需要 ----
# pytorch-lightning
# omegaconf
huggingface_hub