correction

app.py

@@ -7,24 +7,22 @@ import numpy as np
 from PIL import Image
 
 # --- Load Your Model and Class Names ---
-# Use st.cache_resource to load the model only once
 @st.cache_resource
 def load_my_model():
-    # Make sure this file name matches your model file
-    model = load_model('resnet50_dryfruits.keras') 
+    # Load the new, modern .keras file
+    model = load_model('resnet50_dryfruits_best.keras') 
     return model
 
-# --- This is the updated dictionary based on your list ---
 class_names = {
     0: 'AlmondGrade_A',
-    1: 'CashewGrade_B',
-    2: 'RaisinGrade_A',
-    3: 'CashewGrade_A',
-    4: 'AlmondGrade_B',
+    1: 'AlmondGrade_B',
+    2: 'CashewGrade_A',
+    3: 'CashewGrade_B',
+    4: 'CashewGrade_C',
     5: 'PistachioGrade_A',
-    6: 'RaisinGrade_B',
-    7: 'WalnutGrade_A',
-    8: 'CashewGrade_C'
+    6: 'RaisinGrade_A',
+    7: 'RaisinGrade_B',
+    8: 'WalnutGrade_A'
 }
 # --------------------------------------------------------