Commit
·
ccb1271
1
Parent(s):
5847a94
Changing name of model from export.pkl to model.pkl
Browse filesNecessary fro creating a good downstream instruction
- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -11,7 +11,7 @@ class PreTrainedPipeline:
|
|
| 11 |
# Preload all the elements you are going to need at inference.
|
| 12 |
# For instance your model, processors, tokenizer that might be needed.
|
| 13 |
# This function is only called once, so do all the heavy processing I/O here"""
|
| 14 |
-
self.model = load_learner(os.path.join(path, "
|
| 15 |
with open(os.path.join(path, "config.json")) as config:
|
| 16 |
config = json.load(config)
|
| 17 |
self.id2label = config["id2label"]
|
|
|
|
| 11 |
# Preload all the elements you are going to need at inference.
|
| 12 |
# For instance your model, processors, tokenizer that might be needed.
|
| 13 |
# This function is only called once, so do all the heavy processing I/O here"""
|
| 14 |
+
self.model = load_learner(os.path.join(path, "model.pkl"))
|
| 15 |
with open(os.path.join(path, "config.json")) as config:
|
| 16 |
config = json.load(config)
|
| 17 |
self.id2label = config["id2label"]
|