transformers
timm
torch
gradio
matplotlib
albumentations