import gradio as gr from PIL import Image import os import torch import torch.nn.functional as F import torchvision.transforms as transforms import torchvision import numpy as np import yaml from huggingface_hub import hf_hub_download from ultralytics import YOLO