stanford-nlpxed
/

transcript-analysis

Inference Endpoints

Model card Files Files and versions Community

ddemszky commited on May 16, 2023

Commit

3a7acca

•

1 Parent(s): 8c699d3

changed path

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -197,7 +197,7 @@ class UptakeModel:
 class EndpointHandler():
-    def __init__(self, , path="."):
         print("Loading models...")
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")

 class EndpointHandler():
+    def __init__(self, path="."):
         print("Loading models...")
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")