fix path to model

handler.py

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