annanau commited on
Commit
a687202
·
verified ·
1 Parent(s): 31dc0b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +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-head.h5")
10
 
11
  # Define the labels for your classification
12
  class_labels = ['fresh', 'early decay', 'advanced decay','skeletonized'] # 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-head")
10
 
11
  # Define the labels for your classification
12
  class_labels = ['fresh', 'early decay', 'advanced decay','skeletonized'] # Replace with your actual class names