Update app.py

app.py

@@ -1,7 +1,23 @@
+import os
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
 import tensorflow as tf
+import cv2
+import imghdr
 import os
-# Avoid OOM errors by setting GPU Memory Consumption Growth
-gpus = tf.config.experimental.list_physical_devices('GPU')
-for gpu in gpus: 
-    tf.config.experimental.set_memory_growth(gpu, True)
 
+# Clean image files
+data_dir = 'data'
+image_exts = ['jpeg', 'jpg', 'bmp', 'png']
+
+for image_class in os.listdir(data_dir):
+    for image in os.listdir(os.path.join(data_dir, image_class)):
+        image_path = os.path.join(data_dir, image_class, image)
+        try:
+            img = cv2.imread(image_path)
+            tip = imghdr.what(image_path)
+            if tip not in image_exts:
+                print('Removing:', image_path)
+                os.remove(image_path)
+        except Exception as e:
+            print('Issue with image:', image_path)