Update app.py

app.py

@@ -11,6 +11,7 @@ import cloudpickle
 from transformers import DistilBertTokenizerFast
 import os
 from logger import get_logger
+import gc
 from find_similar_news import TextVectorizer, get_milvus_collection, load_sentence_transformer
 
 app = Flask(__name__)
@@ -70,6 +71,7 @@ def update_news():
         logger.warning(f'ERROR IN update_news(): {e}')
     
     logger.warning('Exiting update_news()')
+    gc.collect()
     return Response(status_json, status=status_code, mimetype='application/json')
 
 logger.warning('Exiting application')