Update app.py

app.py

@@ -7,6 +7,13 @@ from tensorflow.keras.layers import Dense, GlobalAveragePooling2D, Dropout, Batc
 from tensorflow.keras.models import Model
 from tensorflow.keras.optimizers import Adam
 
+# Function to find the greatest batch size that fully divides the total samples
+def find_batch_size(total_samples, initial_batch_size):
+    for bs in range(initial_batch_size, 0, -1):
+        if total_samples % bs == 0:
+            return bs
+    return 1  # Fallback to 1 if no perfect divisor is found
+
 # Pfad zum Verzeichnis der Bilddaten
 base_dir = 'pokemon-images-first-generation17000-files/pokemon'  # Pfad anpassen