Update app.py

app.py

@@ -11,6 +11,7 @@ import cloudpickle
 from transformers import DistilBertTokenizerFast
 import os
 from logger import get_logger
+from find_similar_news import TextVectorizer, get_milvus_collection, load_sentence_transformer
 
 app = Flask(__name__)
 CORS(app)
@@ -31,6 +32,9 @@ def load_model():
     return interpreter, label_encoder, tokenizer
 
 interpreter, label_encoder, tokenizer = load_model()
+vectorizer = TextVectorizer()
+collection = get_milvus_collection()
+sent_model, ce_model = load_sentence_transformer()
 
 
 @app.route("/")