Welcome to the Pneumonia Detection project! This repository contains code for training a Convolutional Neural Network (CNN) with attention mechanism to detect pneumonia from chest X-ray images. Below, you'll find information on the dataset used for training, how to access the trained model, and where to find the hosted application.
For training the pneumonia detection model, the Chest X-ray Images (Pneumonia) dataset from Kaggle was used. This dataset contains X-ray images of healthy and pneumonia-infected lungs.
The trained model is hosted on the Hugging Face Spaces platform. You can access the application here. The application allows you to upload a chest X-ray image and get predictions on whether pneumonia is detected.
Due to the large size of the model files (2.5GB each), it's recommended to access the model through the Hugging Face hosted space. You can download the model from the provided link in the application space.
- Clone this repository to your local machine:
git clone https://github.com/your-username/Pneumonia-detection-Using-CNN-attension.git- Install the required dependencies:
pip install -r requirements.txt- Run the application script:
python app.py- Navigate to
http://localhost:5000in your web browser to access the application locally.
Contributions to improve the model, add features to the application, or fix bugs are welcome! Please open an issue to discuss proposed changes or submit a pull request directly.
This project is licensed under the MIT License - see the LICENSE file for details.
👨💻 Happy pneumonia detection! If you have any questions or need assistance, feel free to reach out.