Update app.py

app.py

@@ -4,13 +4,12 @@ from PIL import Image
 import numpy as np
 
 # Load your custom regression model
-model_path = "kia_mnist_keras_model.weights.h5"
-model_path = "kia_mnist_keras_model.keras"
+model_path = "pokemon_transferlearning.keras"
 
 #model.load_weights(model_path)
 model = tf.keras.models.load_model(model_path)
 
-labels = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
+labels = ['Porygon', 'Seel', 'Vaporeon']
 
 # Define regression function
 def predict_regression(image):
@@ -26,10 +25,10 @@ def predict_regression(image):
 
 # Create Gradio interface
 input_image = gr.Image()
-output_text = gr.Textbox(label="Predicted Value")
+output_text = gr.Textbox(label="Predicted Pokemon")
 interface = gr.Interface(fn=predict_regression, 
                          inputs=input_image, 
                          outputs=gr.Label(),
-                         examples=["images/0.jpeg", "images/1.jpeg", "images/2.jpeg", "images/5.jpeg"],   
-                         description="A simple mlp classification model for image classification using the mnist dataset.")
+                         examples=["images/porygon.jpeg", "images/seel.jpeg", "images/vaporeon.jpeg"],   
+                         description="A simple mlp classification model for pokemon classification.")
 interface.launch()