espejelomar
commited on
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"]
|