ydshieh commited on
Commit
790ecb1
1 Parent(s): 7a16b57

update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +1 -1
generate.py CHANGED
@@ -68,7 +68,7 @@ if __name__ == '__main__':
68
  print(f'caption: {caption}')
69
 
70
  ds = datasets.load_dataset('./coco_dataset_script.py', data_dir='/home/33611/caption/')
71
- ds = ds['train']
72
  ds = ds.select(range(100))
73
 
74
  predictions = []
 
68
  print(f'caption: {caption}')
69
 
70
  ds = datasets.load_dataset('./coco_dataset_script.py', data_dir='/home/33611/caption/')
71
+ ds = ds['validation']
72
  ds = ds.select(range(100))
73
 
74
  predictions = []