Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ from PIL import Image
|
|
6 |
# Modell laden
|
7 |
model = tf.keras.models.load_model('pokemon_classifier_model.h5')
|
8 |
|
9 |
-
class_names = ['
|
10 |
|
11 |
def predict(image):
|
12 |
image = image.resize((150, 150))
|
|
|
6 |
# Modell laden
|
7 |
model = tf.keras.models.load_model('pokemon_classifier_model.h5')
|
8 |
|
9 |
+
class_names = ['Chansey', 'Growlithe', 'Lapras']
|
10 |
|
11 |
def predict(image):
|
12 |
image = image.resize((150, 150))
|