Update app.py

app.py

@@ -25,7 +25,7 @@ def predict(image):
     predictions = model.predict(preprocessed)
     
     # Define the class labels (you can update these with your actual class names)
-    class_labels = ['Class 1', 'Class 2', 'Class 3', 'Class 4', 'Class 5']
+    class_labels = ['No DR', 'Mild', 'Moderate', 'Severe', 'Proliferative DR']
     
     # Create a dictionary with class labels as keys and probabilities as values
     prediction_dict = {class_labels[i]: float(predictions[0][i]) for i in range(len(class_labels))}