fix
This commit is contained in:
parent
968f94b4b8
commit
b7737efe0c
10
main.py
10
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,'
|
||||
|
|
Loading…
Reference in New Issue