stanford-nlpxed
/

transcript-analysis

Model card Files Files and versions

ddemszky commited on May 16, 2023

Commit

8c699d3

·

1 Parent(s): 6721c85

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):
         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")