requirements.txt

# Remove the --extra-index-url line below if you're not using Nvidia GPU.
# If you're using it, Update CUDA version to yours :
# For CUDA 12.1, use : https://download.pytorch.org/whl/cu121
# For CUDA 12.4, use : https://download.pytorch.org/whl/cu124
--extra-index-url https://download.pytorch.org/whl/cu124


torch
torchvision
pytest
gradio
git+https://github.com/jhj0517/gradio-image-prompter.git
opencv-python
pycocotools
git+https://github.com/jhj0517/segment-anything-2.git
git+https://github.com/jhj0517/forked-pytoshop.git
packbits
hydra-core
numpy==1.26.4
wheel