waste-classifier / requirements.txt
santit96's picture
Stop versioning the model checkpoints, now they are downloaded from huggingface. Add env vars
dd14920
albumentations>=0.5.2
efficientnet_pytorch
fastai==2.7.13
funcy==1.15
huggingface_hub
iterative-stratification==0.1.6
matplotlib==3.8.2
numpy==1.26.2
omegaconf>=2.0
opencv-python==4.8.1.78
opencv-python-headless==4.8.1.78
pycocotools>=2.0.0
python-dotenv
pytorch_lightning
pyyaml
rembg==2.0.53
scikit-learn==1.3.2
scikit-plot
scipy==1.11.4
streamlit
timm
torch
torchvision
tqdm==4.66.1