Update app.py

app.py

@@ -12,7 +12,7 @@ class_names = ['Aerodactyl', 'Charizard', 'Victreebel']
 
 def classify_image(image):
     image = Image.fromarray(image.astype('uint8'), 'RGB')
-    img = image.resize((150, 150))
+    img = image.resize((224, 224))  # Ändern Sie die Größe auf 224x224
     img_array = tf.keras.preprocessing.image.img_to_array(img)
     img_array = tf.expand_dims(img_array, 0)  # Erstelle einen Batch
     predictions = model.predict(img_array)
@@ -20,6 +20,7 @@ def classify_image(image):
     confidence = np.max(predictions[0])
     return {predicted_class: float(confidence)}
 
+
 image_input = gr.Image()
 label = gr.Label(num_top_classes=3)