Update app.py

app.py

@@ -3,9 +3,11 @@ import tensorflow as tf
 from tensorflow.keras.preprocessing import image
 import numpy as np
 from PIL import Image
+from keras import layers
 
 # Load your trained Xception model
-model = tf.keras.models.load_model("xception-070523")
+#model = tf.keras.models.load_model("xception-070523")
+model = layers.TFSMLayer("xception-070523", call_endpoint='serving_default')
 
 # Define the labels for your classification (example: if you have 3 classes)
 class_labels = ['0', '1', '2', '3']  # Replace with your actual class names