osanseviero commited on
Commit
5694969
·
1 Parent(s): deba7ca

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -4,7 +4,7 @@ from typing import Dict
4
  from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
5
  from pyctcdecode import Alphabet, BeamSearchDecoderCTC
6
 
7
- class PreTrainedModel():
8
  def __init__(self, path):
9
  """
10
  Initialize model
 
4
  from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
5
  from pyctcdecode import Alphabet, BeamSearchDecoderCTC
6
 
7
+ class PreTrainedPipeline():
8
  def __init__(self, path):
9
  """
10
  Initialize model