Update app.py

app.py

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