Updated lines 22 and 44 with: Jolteon, Kakuna, and Mr. Mime

app.py

@@ -19,7 +19,7 @@ model = tf.keras.models.load_model('pokemon_model.keras')
 # Los nombres de las clases deben coincidir con su conjunto de datos.
 # Klassennamen, sollten deinem Dataset entsprechen
 # 类名应该与你的数据集匹配
-class_names = ['Gyarados', 'Primape', 'Wynaut']
+class_names = ['Jolteon', 'Kakuna', 'Mr. Mime']
 
 
 def classify_image(image):
@@ -41,6 +41,6 @@ iface = gr.Interface(
     inputs=image_input,
     outputs=label,
     title='Pokémon Classifier',
-    description='Upload an image of Gyarados, Primape, or Wynaut and the classifier will tell you which one it is and the confidence level of the prediction.').launch()
+    description='Upload an image of Jolteon, Kakuna, or Mr. Mime and the classifier will tell you which one it is and the confidence level of the prediction.').launch()
 
 #iface.launch()