gradio
torch
transformers
yolov5
os