bauckluc commited on
Commit
a4e4ac9
·
verified ·
1 Parent(s): c48b1e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ from PIL import Image
4
  import numpy as np
5
 
6
  # Load your custom regression model
 
7
  model_path = "pokemon_transferlearning.keras"
8
 
9
  model = tf.keras.models.load_model(model_path)
@@ -31,4 +32,4 @@ interface = gr.Interface(fn=predict_regression,
31
  outputs=gr.Label(),
32
  examples=["images/porygon.png", "images/seel.jpg", "images/vaporeon.png"],
33
  description="A simple mlp classification model for pokemon classification.")
34
- interface.launch()
 
4
  import numpy as np
5
 
6
  # Load your custom regression model
7
+ model_path = "kia_mnist_keras_model.weights.h5"
8
  model_path = "pokemon_transferlearning.keras"
9
 
10
  model = tf.keras.models.load_model(model_path)
 
32
  outputs=gr.Label(),
33
  examples=["images/porygon.png", "images/seel.jpg", "images/vaporeon.png"],
34
  description="A simple mlp classification model for pokemon classification.")
35
+ interface.launch()