Update app.py

app.py

@@ -10,6 +10,7 @@ model = tf.keras.models.load_model('pokemon_model.keras')
 class_names = ['Jolteon', 'Kakuna', 'Mr. Mime']
 
 def classify_image(image):
+    image = Image.fromarray(image.astype('uint8'), 'RGB')
     img = image.resize((160, 160))
     img_array = tf.keras.preprocessing.image.img_to_array(img)
     img_array = tf.expand_dims(img_array, 0)  # Erstelle einen Batch
@@ -18,8 +19,9 @@ def classify_image(image):
     confidence = np.max(predictions[0])
     return predicted_class, confidence
 
-image_input = gr.Image(shape=(160, 160)) 
-label = gr.Label(num_top_classes=3)       
+
+image_input = gr.Image()
+label = gr.Label(num_top_classes=3)      
 
 iface = gr.Interface(
     fn=classify_image,