Update app.py

app.py

@@ -16,7 +16,7 @@ def predict_pokemon(img):
     img_array = preprocess_input(img_array)  # Preprocess the input as expected by ResNet50
     
     prediction = model.predict(img_array)  # Predict using the model
-    classes = ['Charmeleon', 'Dewgong', 'Zubat' ]  # Specific Pokémon names
+    classes = ['Arbok', 'Slowpoke', 'Zapdos' ]  # Specific Pokémon names
     return {classes[i]: float(prediction[0][i]) for i in range(3)}  # Return the prediction
 
 # Define Gradio interface