The model is a Convolutional Neural Network (CNN) trained on the retinal images to predict the disease.
Dataset used for training is MEDMNIST dataset which contains retinal images of 4 different diseases.
It uses PyTorch framework for training and prediction.
Upload an image of the retina and click on 'Submit' to get the prediction.
It will show the predicted disease based on the input image.
It can predict one of the following diseases:
The model has an accuracy of 75% on the test dataset.