mae-waste-classifier-demo / requirements.txt
ysfad's picture
Upload requirements.txt with huggingface_hub
18b5f98 verified
# Core ML libraries (CPU-only for HF Spaces)
torch>=2.0.0,<3.0.0 --index-url https://download.pytorch.org/whl/cpu
torchvision>=0.15.0,<1.0.0 --index-url https://download.pytorch.org/whl/cpu
open_clip_torch>=2.20.0,<3.0.0
# Hugging Face integration
huggingface_hub>=0.19.0,<1.0.0
# Image processing
pillow>=9.0.0,<11.0.0
# Data processing
pandas>=1.5.0,<3.0.0
numpy>=2.0.0
# API & UI framework
pydantic==2.10.6
gradio
requests
python-dotenv
tqdm
torchaudio
timm
transformers