requirements.txt

# Remove --extra-index-url URL below if you're not using nvidia GPU.
# If you're using it, Update CUDA version to yours. For example /cu121 , /cu118 etc..
--extra-index-url https://download.pytorch.org/whl/cu124
torch
torchvision
pytest
gradio
gradio_image_prompter
opencv-python
pycocotools
git+https://github.com/jhj0517/forked-pytoshop.git
packbits
hydra-core
numpy==1.26.4
wheel