torch numpy opencv-python pickle gradio tqdm matplotlib grad-cam scikit-learn