Create script.py

script.py

@@ -0,0 +1,32 @@
+import os
+from PIL import Image
+import pandas as pd
+
+def get_image_data(folder_path):
+    image_data = []
+    
+    # Iterate through all files in the folder
+    for filename in os.listdir(folder_path):
+        file_path = os.path.join(folder_path, filename)
+        
+        # Check if the file is an image
+        if os.path.isfile(file_path) and filename.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.bmp')):
+            try:
+                with Image.open(file_path) as img:
+                    width, _ = img.size
+                    image_data.append({'file_name': filename, 'image_width': width})
+            except IOError:
+                print(f"Error opening {filename}. Skipping this file.")
+    
+    return image_data
+
+# Specify the folder path
+folder_path = "/tmp/data/hugging_face_images"
+
+# Get image data
+image_data = get_image_data(folder_path)
+
+# Create DataFrame
+df = pd.DataFrame(image_data)
+
+df.to_csv("solution.csv", index=False)