Update app.py

app.py

@@ -23,7 +23,16 @@ DepthwiseConv2D.from_config = classmethod(
 
 # --- Constants ---
 IMG_SIZE = (224, 224)
-CLASS_NAMES = ['Normal', 'Diabetes', 'Glaucoma', 'Cataract', 'AMD', 'Hypertension', 'Myopia', 'Others']
+CLASS_NAMES = [
+    'Normal', 
+    'Diabetic Retinopathy',   # Changed from 'Diabetes'
+    'Glaucoma', 
+    'Cataract', 
+    'Age-related Macular Degeneration (AMD)',  # Changed from 'AMD'
+    'Hypertension', 
+    'Myopia', 
+    'Others'
+]
 LIME_EXPLAINER = lime_image.LimeImageExplainer()
 
 # --- Load Model ---
@@ -89,7 +98,7 @@ The model predicted <strong>Normal</strong> because everything looks healthy:<br
 - 👁 Disc and macula are well-defined<br>
 <span style='color:green'>✔️ This is a healthy eye.</span>""",
 
-    'Diabetes': """💉 **<span style='color:orange'>Diabetes</span>**<br>
+    'Diabetic Retinopathy': """💉 **<span style='color:orange'>Diabetic Retinopathy</span>**<br>
 Signs of diabetic eye disease:<br>
 - 🔶 Tiny red spots or swelling<br>
 - 🩸 Damaged blood vessels<br>
@@ -110,7 +119,7 @@ Blurry image due to lens clouding:<br>
 - 🔍 Faded details in retina<br>
 <span style='color:gray'>⚠️ Cataract suspected.</span>""",
 
-    'AMD': """🧓 **<span style='color:red'>AMD</span>**<br>
+    'Age-related Macular Degeneration (AMD)': """🧓 **<span style='color:red'>Age-related Macular Degeneration (AMD)</span>**<br>
 Macula shows degeneration:<br>
 - 🔴 Yellow spots in center<br>
 - 👁 Age-related changes<br>
@@ -169,9 +178,6 @@ def show_lime(img, model, pred_idx, pred_label, all_probs):
             st.markdown(explanation_text.get(pred_label, "No explanation available."), unsafe_allow_html=True)
 
 
-
-
-
 # --- Streamlit UI ---
 st.set_page_config(page_title="👁 Retina Classifier with LIME", layout="wide")
 st.title("👁 Retina Disease Classifier with LIME + Confidence Chart")