rajaatif786
/

TickerExtraction

Feature Extraction

Model card Files Files and versions

rajaatif786 commited on Aug 19, 2022

Commit

01b32cf

·

1 Parent(s): fdcadae

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -1

pipeline.py CHANGED Viewed

@@ -159,7 +159,8 @@ class PairRelInstance:
 class PreTrainedPipeline():
     def __init__(self, path):
-        self.model = EntityBertNet()
     def __call__(self, inputs)-> Dict[str, str]:

 class PreTrainedPipeline():
     def __init__(self, path):
+        config = BertConfig.from_pretrained(TRAINED_WEIGHTS)
+        self.model = BertModel.from_pretrained(TRAINED_WEIGHTS, config=config)
     def __call__(self, inputs)-> Dict[str, str]: