🧠 VisionAI - Image Classifier

An AI-powered image classification web application built with Python, PyTorch, and Streamlit.

This project uses a trained deep learning model to classify uploaded images and predict the object category in real time.

Built using a Convolutional Neural Network (CNN) trained with PyTorch for image classification.

---

🚀 Features

• 📸 Upload an image from your computer
• 🧠 AI model predicts the image class
• ⚡ Fast and interactive web interface
• 🌐 Built with Streamlit
• 🐳 Docker-ready project
• 💻 Ready for portfolio and deployment

---

🛠️ Tech Stack

• Python
• PyTorch
• Torchvision
• Streamlit
• Pillow
• Docker

---

📂 Project Structure

image-classifier-ai/
│
├── app.py
├── train_model.py
├── model.pth
├── requirements.txt
├── Dockerfile
├── README.md
└── .gitignore

---

▶️ Run Locally

Install dependencies:

pip install -r requirements.txt

Run the app:

streamlit run app.py

Open in browser:

http://localhost:8501

---

🐳 Run with Docker

Build image:

docker build -t visionai .

Run container:

docker run -p 8501:8501 visionai

---

💡 About

This project was created as part of my learning journey in Artificial Intelligence and Python development.

It combines machine learning with a professional web interface to showcase practical AI and software development skills.

---

👩‍💻 Author

María Agustina AI • Python • Software Development •