ImageNet / requirements.txt
Shilpaj's picture
Refactor: Modifications for inference on GPU
f8ecba6
raw
history blame
83 Bytes
gradio==3.50.2
grad-cam
numpy<2.0.0
torch==2.0.1
torchvision==0.15.2
Pillow>=9.0.0