torch
torchvision
gradio
Pillow
diffusers
segment-anything