Update app.py

app.py

@@ -4,13 +4,13 @@ from PIL import Image
 import numpy as np
 
 # Load your custom regression model
-model_path = "Ditto-premiumdelux-model_transferlearning.weights.h5"
-model_path = "Ditto-premiumdelux-model_transferlearning.keras"
+model_path = "pokemon_model_transferlearning.weights.h5"
+model_path = "pokemon_model_transferlearning.keras"
 
 #model.load_weights(model_path)
 model = tf.keras.models.load_model(model_path)
 
-labels = ['Ditto','Golbat','Koffing']
+labels = ['Ditto','Venomoth','Venusaur']
 
 # Define regression function
 def predict_regression(image):