⚡basic-image-classifier⚡

This project is an image classifier integrating a simple convolutional neural network(CNN), for this the following libraries were used:

• PyTorch (https://pytorch.org/)
• Streamlit (https://streamlit.io/)
• Pillow (https://python-pillow.org/)

video_demo.mp4

🗂 In this project we use PyTorch to create our CNN, we also use it to apply some transformations to the set of images to obtain better results

🚨Prerequisites🚨

• python 3.10 ^

🚧local installation🚧

1. create the virtual environment with pip venv and activate it

python -m venv venv
source venv/bin/activate # linux

2. install dependencies

pip install -r requirements.txt

3. run the app

streamlit run app.py 🚀🚀