espejelomar commited on
Commit
ccb1271
1 Parent(s): 5847a94

Changing name of model from export.pkl to model.pkl

Browse files

Necessary fro creating a good downstream instruction

Files changed (1) hide show
  1. 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, "export.pkl"))
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"]