Spaces:
Sleeping
Sleeping
File size: 835 Bytes
c65837d c695ced c65837d |
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 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import pandas as pd
import json
import gradio as gr
def create_jsonl(csv_file):
df = pd.read_csv(csv_file)
column_list = df.columns
output = 'metadata.jsonl'
total_data = []
for index, row in df.iterrows():
data = {}
for column in column_list:
if column == 'file_name':
file_name = row[column]
data[column] = row[column]
row_data = {"file_name": file_name, "ground_truth": json.dumps(data)}
total_data.append(row_data)
with open(output, 'a', encoding='utf-8') as f:
for row_data in total_data:
f.write(json.dumps(row_data) + '\n')
return output
demo = gr.Interface(fn=create_jsonl,inputs=gr.File(),outputs="text", title=f"jsonl converter",)
demo.launch()
|