Spaces:

reiflja1
/

pokemon-classifier

Runtime error

reiflja1 commited on May 14

Commit

05d40b5

•

1 Parent(s): 592d01e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,8 @@
 import gradio as gr
 import tensorflow as tf
 # Load the model
-model = tf.keras.models.load_model("pokemon_classifier_model.keras")
 def predict(image):
     img = tf.keras.preprocessing.image.img_to_array(image)

 import gradio as gr
 import tensorflow as tf
 # Load the model
+model_path = "pokemon_classifier_model.keras"
+model = tf.keras.models.load_model(model_path)
 def predict(image):
     img = tf.keras.preprocessing.image.img_to_array(image)