Update app.py
Browse files
app.py
CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
|
|
2 |
import tensorflow as tf
|
3 |
import numpy as np
|
4 |
from PIL import Image
|
|
|
5 |
|
6 |
# Load your custom classification model
|
7 |
model = tf.keras.models.load_model('pokemon_classifier_model.keras')
|
|
|
2 |
import tensorflow as tf
|
3 |
import numpy as np
|
4 |
from PIL import Image
|
5 |
+
import keras as keras
|
6 |
|
7 |
# Load your custom classification model
|
8 |
model = tf.keras.models.load_model('pokemon_classifier_model.keras')
|