Update app.py

app.py

@@ -8,7 +8,7 @@ model_path = "pokemon.keras"
 model = tf.keras.models.load_model(model_path)
 model.summary()  # Check if the model architecture loaded matches the expected one
 
-labels = ['Abra', 'Aerodactyl', 'Alakazam', 'Arbok', 'Arcanine']
+labels = ['0','1','2','3','4']
 
 # Define regression function
 def predict_regression(image):