membrfab commited on
Commit
a0733d5
1 Parent(s): bd813af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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')