requirements.txt

torch>=1.9.0
pydantic==2.10.6
torchvision>=0.10.0
gradio==5.39.0
pillow>=8.0.0
numpy>=1.21.0
segmentation-models-pytorch>=0.3.0
opencv-python>=4.5.0
albumentations>=1.3.0
matplotlib>=3.5.0
scikit-learn>=1.1.0
opencv-python>=4.5.0
tqdm>=4.64.0
git+https://github.com/jacobgil/pytorch-grad-cam.git