lalithadevi commited on
Commit
f7374b3
1 Parent(s): c1b9bc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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')