Update app.py

app.py

@@ -5,10 +5,11 @@ import numpy as np
 
 # Load your custom regression model
 model_path = "pokemon.keras"
+model_path = "pokemon.weights.h5"
 model = tf.keras.models.load_model(model_path)
 model.summary()  # Check if the model architecture loaded matches the expected one
 
-labels = ['0','1','2','3','4']
+labels = ['Abra', 'Aerodactyl', 'Alakazam', 'Arbok', 'Arcanine']
 
 # Define regression function
 def predict_regression(image):