From b7737efe0c781a99b5e07cf3a82849c2b24630a7 Mon Sep 17 00:00:00 2001 From: ahmed531998 Date: Tue, 18 Apr 2023 23:09:33 +0200 Subject: [PATCH] fix --- main.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/main.py b/main.py index bbfcf96..99f7bc1 100644 --- a/main.py +++ b/main.py @@ -7,6 +7,7 @@ from flask_cors import CORS, cross_origin import psycopg2 import spacy import requests +import shutil import spacy_transformers import torch from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline @@ -202,6 +203,15 @@ if __name__ == "__main__": threading.Thread(target=clear_inactive, name='clear').start() rec = Recommender(retriever) + for filename in os.listdir("/app/"): + file_path = os.path.join(folder, filename) + try: + if os.path.isfile(file_path) or os.path.islink(file_path): + os.unlink(file_path) + elif os.path.isdir(file_path): + shutil.rmtree(file_path) + except Exception as e: + print('Failed to delete %s. Reason: %s' % (file_path, e)) cur.execute('CREATE TABLE IF NOT EXISTS feedback_experimental (id serial PRIMARY KEY,' 'query text NOT NULL,'