hsalih01 commited on
Commit
04c1695
·
verified ·
1 Parent(s): 48dbf61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ model_path = "pokemon_classifier_model.keras"
8
  model = tf.keras.models.load_model(model_path)
9
 
10
  # Define the Pokémon classes
11
- classes = ['Doduo', 'Geodude', 'Zubat'] # Adjust these as per your model's classes
12
 
13
  # Define the image classification function
14
  def classify_image(image):
 
8
  model = tf.keras.models.load_model(model_path)
9
 
10
  # Define the Pokémon classes
11
+ classes = ['Dragonite', 'Bulbasaur', 'Golbat'] # Adjust these as per your model's classes
12
 
13
  # Define the image classification function
14
  def classify_image(image):