PyTorch, specifically, torchvision has a lot of models and pre-trained weights. CNN Models are mostly trained on ImageNet. Hyperparameters can be found here.
Install the required libraries by running the following command:
pip install -r requirements.txt
Run the streamlit app via the following command:
streamlit run main.py
Upload the image you want to classify.