Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 = ['
|
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):
|