Spaces:
Running
Running
File size: 543 Bytes
2cd207c 6febf41 dbb7d2e 2cd207c b21aab5 2cd207c 4d4f66e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
from datasets import load_dataset, Dataset, load_from_disk
def main():
promptbook = load_dataset('NYUSHPRP/ModelCofferPromptBook', split='train')
print(promptbook)
promptbook.save_to_disk('./promptbook')
#
# roster = load_dataset('NYUSHPRP/ModelCofferRoster', split='train')
# roster.save_to_disk('./roster')
def load():
roster = load_from_disk('./roster')
print(roster)
def test():
promptbook = load_from_disk('./promptbook')
print(promptbook[0]['image'])
if __name__ == '__main__':
main()
|