reiflja1 commited on
Commit
9b8ce47
1 Parent(s): 8a458c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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('pokemon_classifier_model.keras')
5
 
6
  def predict(image):
7
  img = tf.keras.preprocessing.image.img_to_array(image)
@@ -21,4 +21,4 @@ def predict(image):
21
  iface = gr.Interface(fn=predict, inputs=gr.Image(), outputs="text", title="Pokémon Classifier")
22
 
23
  # Run the interface
24
- iface.launch()
 
1
  import gradio as gr
2
  import tensorflow as tf
3
  # Load the model
4
+ model = tf.keras.models.load_model('/home/user/app/pokemon_classifier_model.keras')
5
 
6
  def predict(image):
7
  img = tf.keras.preprocessing.image.img_to_array(image)
 
21
  iface = gr.Interface(fn=predict, inputs=gr.Image(), outputs="text", title="Pokémon Classifier")
22
 
23
  # Run the interface
24
+ iface.launch()