osanseviero commited on
Commit
d34ffad
1 Parent(s): 49fb52d

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -7,7 +7,7 @@ class PreTrainedPipeline():
7
  # For instance your model, processors, tokenizer that might be needed.
8
  # This function is only called once, so do all the heavy processing I/O here"""
9
  raise NotImplementedError(
10
- "Please implement TokenClassificationPipeline __init__ function"
11
  )
12
 
13
  def __call__(self, inputs: str) -> List[Dict[str, Any]]:
@@ -25,5 +25,5 @@ class PreTrainedPipeline():
25
  """
26
  # IMPLEMENT_THIS
27
  raise NotImplementedError(
28
- "Please implement TokenClassificationPipeline __call__ function"
29
  )
 
7
  # For instance your model, processors, tokenizer that might be needed.
8
  # This function is only called once, so do all the heavy processing I/O here"""
9
  raise NotImplementedError(
10
+ "Please implement PreTrainedPipeline __init__ function"
11
  )
12
 
13
  def __call__(self, inputs: str) -> List[Dict[str, Any]]:
 
25
  """
26
  # IMPLEMENT_THIS
27
  raise NotImplementedError(
28
+ "Please implement PreTrainedPipeline __call__ function"
29
  )