Update app.py

app.py

@@ -5,7 +5,6 @@ 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
 
@@ -30,7 +29,7 @@ def predict_regression(image):
 # Create Gradio interface
 input_image = gr.Image()
 output_text = gr.Textbox(label="Predicted Value")
-interface = gr.Interface(fn=predict_regression, 
+interface = gr.Interface(fn=predict_classification, 
                          inputs=input_image, 
                          outputs=gr.Label(),
                          examples=["images/abra.gif", "images/Aerodactyl.png", "images/Alakazam.png", "images/arbok.jpg", "images/Arcanine.png" ],