Update main.py

main.py

@@ -1,16 +1,13 @@
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 from fastapi.responses import FileResponse
-from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
+from transformers import pipeline
 from pysbd import Segmenter
 
 app = FastAPI()
 segmenter = Segmenter(language='en', clean=False)
-model = AutoModelForSequenceClassification.from_pretrained("remzicam/privacy_intent")
-tokenizer = AutoTokenizer.from_pretrained("mukund/privbert")
 privacy_intent_pipe = pipeline("text-classification",
-                                model = model,
-                                tokenizer = tokenizer)
+                               "remzicam/privacy_intent" )
 
 def doc2sent(text:str)-> dict:
     """