Update main.py

main.py

@@ -4,14 +4,19 @@ from contextlib import asynccontextmanager
 import logging
 from api.endpoints import location
 
+app = FastAPI(lifespan=lifespan)
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    print("Initialising nlp spacy libs")
-    logging.info("Initialising nlp spacy libs")
-    initialize_nlp()
+    # initialize_nlp()
+    print("Initializing NER model and tokenizer")
+    logging.info("Initializing NER model and tokenizer")
+    
+    app.tokenizer = AutoTokenizer.from_pretrained("ml6team/bert-base-uncased-city-country-ner")
+    app.model = AutoModelForTokenClassification.from_pretrained("ml6team/bert-base-uncased-city-country-ner")
+    app.nlp = pipeline('ner', model=app.model, tokenizer=app.tokenizer, aggregation_strategy="simple")
+    
     yield
 
-app = FastAPI(lifespan=lifespan)
 app.include_router(location.router, prefix="/location/api/v1")