Update app.py

app.py

@@ -8,8 +8,8 @@ from keras import layers
 # Load your trained Xception model
 model = tf.keras.models.load_model("xception-head.h5")
 
-# Define the labels for your classification (example: if you have 3 classes)
-class_labels = ['1', '2', '3','4','5','6']  # Replace with your actual class names
+# Define the labels for your classification
+class_labels = ['fresh', 'early decay', 'advanced decay','skeletonized']  # Replace with your actual class names
 
 def classify_image(img):
     # Preprocess the image to fit the model input shape