torch timm numpy pillow einops safetensors gradio requests