annanau commited on
Commit
53e415e
·
verified ·
1 Parent(s): 43fbc49

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,10 +5,10 @@ import numpy as np
5
  from PIL import Image
6
 
7
  # Load your trained Xception model
8
- model = tf.keras.models.load_model("your_xception_model.h5")
9
 
10
  # Define the labels for your classification (example: if you have 3 classes)
11
- class_labels = ['class1', 'class2', 'class3'] # Replace with your actual class names
12
 
13
  def classify_image(img):
14
  # Preprocess the image to fit the model input shape
 
5
  from PIL import Image
6
 
7
  # Load your trained Xception model
8
+ model = tf.keras.models.load_model("xception-070523")
9
 
10
  # Define the labels for your classification (example: if you have 3 classes)
11
+ class_labels = ['0', '1', '2', '3'] # Replace with your actual class names
12
 
13
  def classify_image(img):
14
  # Preprocess the image to fit the model input shape