Omkar008 commited on
Commit
daa4628
·
verified ·
1 Parent(s): 42f800d

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -4,7 +4,7 @@ from contextlib import asynccontextmanager
4
  import logging
5
  from api.endpoints import location
6
  from transformers import AutoTokenizer, AutoModelForTokenClassification
7
-
8
  @asynccontextmanager
9
  async def lifespan(app: FastAPI):
10
  # initialize_nlp()
 
4
  import logging
5
  from api.endpoints import location
6
  from transformers import AutoTokenizer, AutoModelForTokenClassification
7
+ from transformers import pipeline
8
  @asynccontextmanager
9
  async def lifespan(app: FastAPI):
10
  # initialize_nlp()