leenaalsalhi/ConvNeXt-Image-Classifier

ConvNeXt Lung Disease Classifier

This app uses a ConvNeXt model trained for lung disease classification. You can upload a lung X-ray image, and the model will classify it into one of the following categories:

• —Encapsulated Lesions
• —High Density
• —Infectious Diseases
• —Inflammatory Pneumonia
• —Low Density
• —Mediastinal Changes
• —Normal
• —Obstructive Diseases
• —Thoracic Changes

How to Use

1. 1.Upload a lung X-ray image.
2. 2.The model will predict the disease category of the image.

Requirements

The app uses the following libraries:

• —PyTorch
• —Gradio
• —timm
• —torchvision

For running this app, ensure these libraries are installed (as defined in requirements.txt).