HCKLab
/

BiBert-MultiTask-1

Text Classification

Generated from Trainer

text-embeddings-inference

Model card Files Files and versions

Metrics Training metrics Community

Kalaoke commited on Oct 24, 2022

Commit

943a701

·

1 Parent(s): 72528aa

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -217,7 +217,7 @@ class EndpointHandler():
     )
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    tokenizer = AutoTokenizer.from_pretrained(checkpoint)
     tasks = [
         Task(id=0, name='label_classification', type='seq_classification', num_labels=5),

     )
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    tokenizer = AutoTokenizer.from_pretrained(path)
     tasks = [
         Task(id=0, name='label_classification', type='seq_classification', num_labels=5),