Narya-ai
/

lr-bert-base-uncased

Inference Endpoints

Model card Files Files and versions Community

Uan Sholanbayev commited on Oct 13, 2023

Commit

9edfae2

•

1 Parent(s): 9ef518f

added model

Files changed (2) hide show

bert_lr.pkl +0 -0
handler.py +1 -1

bert_lr.pkl ADDED Viewed

Binary file (6.86 kB). View file

handler.py CHANGED Viewed

@@ -16,7 +16,7 @@ def unpickle_obj(filepath):
 class EndpointHandler():
     def __init__(self, path=""):
-        self.model = unpickle_obj(path)
         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(os.getcwd()).to(self.device)

 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.device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
         self.bert = BertModel.from_pretrained(os.getcwd()).to(self.device)