Update app.py

app.py

@@ -3,7 +3,7 @@ import tensorflow as tf
 from PIL import Image
 import numpy as np
 
-# Laden des vortrainierten Modells
+# Laden des vortrainierten Pokémon-Modells
 model_path = "kia_pokemon_keras_model.h5"
 model = tf.keras.models.load_model(model_path)
 
@@ -21,6 +21,7 @@ def predict_pokemon(image):
     image = image.reshape(1, 224, 224, 3)
 
     # Vorhersage treffen
+
     predictions = model.predict(image)
     prediction = np.argmax(predictions, axis=1)[0]
     confidence = np.max(predictions)
@@ -30,7 +31,7 @@ def predict_pokemon(image):
     return result
 
 # Erstellen der Gradio-Oberfläche
-input_image = gr.Image(shape=(224, 224), image_mode='RGB', tool="editor")  # Image-Tool auf "editor" gesetzt für Anpassungen
+input_image = gr.Image(width=224, height=224, image_mode='RGB') 
 output_label = gr.Label()
 interface = gr.Interface(fn=predict_pokemon,
                          inputs=input_image,