annanau commited on
Commit
bf3b200
·
verified ·
1 Parent(s): 75d623a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ from PIL import Image
6
  from keras import layers
7
 
8
  # Load your trained Xception model
9
- #model = tf.keras.models.load_model("xception-070523")
10
- model = layers.TFSMLayer("xception-070523", call_endpoint='serving_default')
11
 
12
  # Define the labels for your classification (example: if you have 3 classes)
13
  class_labels = ['0', '1', '2', '3'] # Replace with your actual class names
 
6
  from keras import layers
7
 
8
  # Load your trained Xception model
9
+ #model = tf.keras.models.load_model("xception-070523.h5")
 
10
 
11
  # Define the labels for your classification (example: if you have 3 classes)
12
  class_labels = ['0', '1', '2', '3'] # Replace with your actual class names