This commit is contained in:
Daniel Hládek 2024-04-11 12:25:17 +02:00
parent b666783b35
commit d5419dffc9
2 changed files with 2 additions and 3 deletions

View File

@ -16,10 +16,8 @@ if __name__ == '__main__':
try:
with open(os.path.join('dataset', 'final_id_v2.jsonl'), encoding='utf-8') as file:
data = [json.loads(line) for line in file]
db.session.add(Samples(data['text']))
for sample in data:
db.session.add(Samples(sample['text']))
db.session.commit()
print('Data sucessfully inserted')
except FileNotFoundError as err:

View File

@ -11,3 +11,4 @@ SQLAlchemy==2.0.28
typing_extensions==4.10.0
Werkzeug==3.0.1
zipp==3.18.1
psycopg2-binary