reiflja1 commited on
Commit
413070f
1 Parent(s): 954d9df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  import tensorflow as tf
3
  # Load the model
4
- model = tf.keras.models.load_model('test.keras')
5
 
6
  def predict(image):
7
  img = tf.keras.preprocessing.image.img_to_array(image)
 
1
  import gradio as gr
2
  import tensorflow as tf
3
  # Load the model
4
+ model = tf.keras.models.load_model('pokemon_classifier_model.keras')
5
 
6
  def predict(image):
7
  img = tf.keras.preprocessing.image.img_to_array(image)