bugfix

.gitignore

@@ -1 +1,2 @@
-pokemon_classifier_model.keras
+pokemon_classifier_model.keras
+main.ipynb

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
-from tensorflow.keras.models import load_model
+import tensorflow as tf
 from PIL import Image
 import numpy as np
 
-model = load_model('pokemon_classifier_model.keras') 
+model_path = "pokemon_classifier_model.keras"
+
+model = tf.keras.models.load_model(model_path)
+
 
 def predict_image(img):
     img = img.resize((224, 224)) 
@@ -19,7 +22,7 @@ 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"],
+                     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.launch()