torch 
torchvision
loguru
gradio