gradio
transformers
torch
pillow
salesforce-lavis==1.0.2
numpy
torchvision
matplotlib