Update app.py

app.py

@@ -8,12 +8,12 @@ model_path = "pokemon_classifier_model.keras"
 model = tf.keras.models.load_model(model_path)
 
 # Define the Pokémon classes
-classes = ['Doduo', 'Geodude', 'Zubat']  # Update these as per your model's classes
+classes = ['Doduo', 'Geodude', 'Zubat']
 
 # Define the image classification function
 def classify_image(image):
     try:
-        # Ensure image is in RGB
+        # Ensure the image is in RGB
         if image.mode != 'RGB':
             image = image.convert('RGB')
         image = image.resize((150, 150))  # Resize to the input size your model expects
@@ -30,8 +30,8 @@ def classify_image(image):
         return str(e)  # Return the error message if something goes wrong
 
 # Create Gradio interface
-input_image = gr.Image(shape=(150, 150))  # Correct way to define inputs
-output_label = gr.Label()  # Correct way to define outputs
+input_image = gr.Image()  # Correct usage without shape parameter
+output_label = gr.Label()
 
 interface = gr.Interface(fn=classify_image,
                          inputs=input_image,
@@ -39,4 +39,4 @@ interface = gr.Interface(fn=classify_image,
                          examples=["path/to/example1.jpg", "path/to/example2.jpg"],  # Optional: Add example images if available
                          description="Upload an image of a Pokémon to classify!")
 
-interface.launch()
+interface.launch()