taquynhnga commited on
Commit
fc22191
1 Parent(s): 738e754

Update pages/4_ImageNet1k.py

Browse files
Files changed (1) hide show
  1. pages/4_ImageNet1k.py +1 -1
pages/4_ImageNet1k.py CHANGED
@@ -41,7 +41,7 @@ with col2:
41
  if submitted:
42
  try:
43
  img_index = int(img_index)
44
- if img_index > 50000 or img_index < 0:
45
  error_container.error('The Image ID must be in range from 0 to 50000', icon="🚫")
46
  else:
47
  image = dataset_dict[img_index//10_000][img_index%10_000]['image']
 
41
  if submitted:
42
  try:
43
  img_index = int(img_index)
44
+ if img_index > 50000-1 or img_index < 0:
45
  error_container.error('The Image ID must be in range from 0 to 50000', icon="🚫")
46
  else:
47
  image = dataset_dict[img_index//10_000][img_index%10_000]['image']