added model

handler.py

@@ -3,7 +3,6 @@ import numpy as np
 from transformers import BertTokenizer, BertModel
 import torch
 import pickle
-import os
 
 
 
@@ -16,10 +15,10 @@ def unpickle_obj(filepath):
 
 class EndpointHandler():
     def __init__(self, path=""):
-        self.model = unpickle_obj(f"{os.getcwd()}/bert_lr.pkl")
-        self.tokenizer = BertTokenizer.from_pretrained(os.getcwd(), local_files_only=True)
+        self.model = unpickle_obj(f"{path}/bert_lr.pkl")
+        self.tokenizer = BertTokenizer.from_pretrained(path, local_files_only=True)
         self.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-        self.bert = BertModel.from_pretrained(os.getcwd()).to(self.device)
+        self.bert = BertModel.from_pretrained(path).to(self.device)
 
     def get_embeddings(self, texts: List[str]):
         inputs = self.tokenizer(texts, return_tensors='pt', truncation=True,