streiluc commited on
Commit
7e10c95
·
verified ·
1 Parent(s): c8a1d50

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import numpy as np
6
  # Load your custom regression model
7
  model_path = "pokemon.keras"
8
  model = tf.keras.models.load_model(model_path)
 
9
 
10
  labels = ['Aerodactyl', 'Arbok', 'Alakazam', 'Abra', 'Arcanine']
11
 
 
6
  # Load your custom regression model
7
  model_path = "pokemon.keras"
8
  model = tf.keras.models.load_model(model_path)
9
+ model.summary() # Check if the model architecture loaded matches the expected one
10
 
11
  labels = ['Aerodactyl', 'Arbok', 'Alakazam', 'Abra', 'Arcanine']
12