Segizu commited on
Commit
38ec03b
·
1 Parent(s): faa12b9

metadata v9

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -44,6 +44,8 @@ def build_database():
44
  # Debug: Print dataset structure
45
  print("Dataset structure:", dataset.features)
46
  print("First item structure:", dataset[0])
 
 
47
 
48
  for i in range(0, len(dataset), batch_size):
49
  batch = dataset[i:i + batch_size]
@@ -51,8 +53,17 @@ def build_database():
51
 
52
  for j, item in enumerate(batch):
53
  try:
54
- # The image is already a PIL Image object
55
- img = item["image"]
 
 
 
 
 
 
 
 
 
56
 
57
  # Ensure image is in RGB mode
58
  img = img.convert("RGB")
 
44
  # Debug: Print dataset structure
45
  print("Dataset structure:", dataset.features)
46
  print("First item structure:", dataset[0])
47
+ print("Dataset type:", type(dataset))
48
+ print("Dataset item type:", type(dataset[0]))
49
 
50
  for i in range(0, len(dataset), batch_size):
51
  batch = dataset[i:i + batch_size]
 
53
 
54
  for j, item in enumerate(batch):
55
  try:
56
+ print(f"Debug - Processing item {i+j}")
57
+ print(f"Debug - Item type: {type(item)}")
58
+ print(f"Debug - Item content: {item}")
59
+
60
+ # Handle both dictionary and direct image access
61
+ if isinstance(item, dict):
62
+ img = item["image"]
63
+ else:
64
+ img = item.image
65
+
66
+ print(f"Debug - Image type: {type(img)}")
67
 
68
  # Ensure image is in RGB mode
69
  img = img.convert("RGB")