From d5419dffc9f900d268de2621a879837538cece7b Mon Sep 17 00:00:00 2001 From: Daniel Hladek Date: Thu, 11 Apr 2024 12:25:17 +0200 Subject: [PATCH] db uri --- annotation_app/initial.py | 4 +--- annotation_app/requirements.txt | 1 + 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/annotation_app/initial.py b/annotation_app/initial.py index f3098fe..2f900dd 100644 --- a/annotation_app/initial.py +++ b/annotation_app/initial.py @@ -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: diff --git a/annotation_app/requirements.txt b/annotation_app/requirements.txt index a490d85..5bf2d49 100644 --- a/annotation_app/requirements.txt +++ b/annotation_app/requirements.txt @@ -11,3 +11,4 @@ SQLAlchemy==2.0.28 typing_extensions==4.10.0 Werkzeug==3.0.1 zipp==3.18.1 +psycopg2-binary