Commit
•
f7374b3
1
Parent(s):
c1b9bc4
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ import cloudpickle
|
|
11 |
from transformers import DistilBertTokenizerFast
|
12 |
import os
|
13 |
from logger import get_logger
|
|
|
14 |
from find_similar_news import TextVectorizer, get_milvus_collection, load_sentence_transformer
|
15 |
|
16 |
app = Flask(__name__)
|
@@ -70,6 +71,7 @@ def update_news():
|
|
70 |
logger.warning(f'ERROR IN update_news(): {e}')
|
71 |
|
72 |
logger.warning('Exiting update_news()')
|
|
|
73 |
return Response(status_json, status=status_code, mimetype='application/json')
|
74 |
|
75 |
logger.warning('Exiting application')
|
|
|
11 |
from transformers import DistilBertTokenizerFast
|
12 |
import os
|
13 |
from logger import get_logger
|
14 |
+
import gc
|
15 |
from find_similar_news import TextVectorizer, get_milvus_collection, load_sentence_transformer
|
16 |
|
17 |
app = Flask(__name__)
|
|
|
71 |
logger.warning(f'ERROR IN update_news(): {e}')
|
72 |
|
73 |
logger.warning('Exiting update_news()')
|
74 |
+
gc.collect()
|
75 |
return Response(status_json, status=status_code, mimetype='application/json')
|
76 |
|
77 |
logger.warning('Exiting application')
|