matplotlib gradio torch transformers numpy