requirements.txt

pytorch-gradcam==0.2.1
pytorch-lightning==1.8.6
pytube==15.0.0
torch==2.0.1
torchmetrics==1.0.3
torchsummary==1.5.1
torchvision==0.15.2
diffusers==0.21.4
ipython==7.28.0
transformers==4.25.1
fastcore==1.5.29
scipy==1.10.0