Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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()
|