An AI-powered Plant Doctor app built with Streamlit and a CNN–Transformer hybrid model. Diagnose leaf diseases in tomato, potato, and pepper crops using deep learning and Grad-CAM visualization. Includes treatment advice via Cohere AI.
Plant Doctor is a powerful Streamlit-based web app that allows users (especially farmers and researchers) to detect and visualize crop leaf diseases using a hybrid CNN–Transformer deep learning model. It supports classification and diagnosis of tomato, potato, and pepper diseases, along with treatment suggestions powered by Cohere’s language model.
Supports 15 common crop conditions including:
Clone the repository
git clone https://github.com/yourusername/plant-doctor-ai.git
cd plant-doctor-ai
Create and activate a virtual environment
python -m venv venv
source venv/bin/activate # or venv\Scripts\activate on Windows
Install requirements
pip install -r requirements.txt
Run the app
streamlit run main.py
Upload a sample tomato leaf image and click Diagnose Leaf Disease to see:
Prediction label: Tomato Late Blight
Confidence: 95.3%
Heatmap showing infection region
Suggested treatment from AI assistant
This project is licensed under the MIT License. See LICENSE for more information.
PlantVillage Dataset
Cohere Language API
TensorFlow, Streamlit, OpenCV, and the ML community 🌍