bugfix

.gitignore

@@ -1,2 +1,3 @@
 pokemon_classifier_model.keras
-main.ipynb
+main.ipynb
+newvenv/*

app.py

@@ -4,25 +4,27 @@ from PIL import Image
 import numpy as np
 
 model_path = "pokemon_classifier_model.keras"
-
 model = tf.keras.models.load_model(model_path)
 
-
 def predict_image(img):
-    img = img.resize((224, 224)) 
+    img = img.resize((224, 224))  # Continue to resize the image as your model expects 224x224
     img = np.array(img)
     img = img / 255.0  
     img = np.expand_dims(img, axis=0)  
 
-    prediction = model.predict(img)  
-    labels = ['Pikachu', 'Sandshrew', 'Squirtle']  
+    prediction = model.predict(img)
+    labels = ['Pikachu', 'Sandshrew', 'Squirtle']
     return {labels[i]: float(prediction[0][i]) for i in range(len(labels))}
 
-iface = gr.Interface(fn=predict_image,
-                     inputs=gr.inputs.Image(shape=(224, 224)),
-                     outputs=gr.outputs.Label(num_top_classes=3),
-                     title="Pokémon Classifier",
-                     examples=["images/pickachu.png", "images/squirtle.png", "images/sandshrew.png", "images/5.jpeg"],
-                     description="Upload an image of Pikachu, Sandshrew, or Squirtle and the classifier will predict which one it is.")
+iface = gr.Interface(
+    fn=predict_image,
+    inputs=gr.components.Image(),  # Corrected usage without the 'shape' parameter
+    outputs=gr.components.Label(num_top_classes=3),
+    title="Pokémon Classifier",
+    examples=["images/pikachu.png", "images/squirtle.png", "images/sandshrew.png"],
+    description="Upload an image of Pikachu, Sandshrew, or Squirtle and the classifier will predict which one it is."
+)
+
+
 
 iface.launch()