Spaces:
Build error
Build error
File size: 747 Bytes
171f55b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import pandas as pd
import json
# read the csv file into a pandas DataFrame
df = pd.read_csv('/root/autodl-tmp/audioset_balanced_train.csv')
captions = df[" caption"].tolist()
df = pd.read_csv('/root/autodl-tmp/Epidemic_all_debiased.csv', on_bad_lines='skip')
captions_2 = df[' caption2'].tolist()
df = pd.read_csv('/root/autodl-tmp/audioset_eval.csv', on_bad_lines='skip')
captions_3 = df['caption'].tolist()
df = pd.read_csv('/root/autodl-tmp/audioset_unbalanced_train.csv', on_bad_lines='skip')
captions_4 = df[' caption'].tolist()
captions = captions + captions_2 + captions_3 + captions_4
print(len(captions))
with open("audio_text.json", "w") as outfile:
# write the data as a JSON string to the file
json.dump(captions, outfile) |