Rachel Bawden
commited on
Commit
•
276d772
1
Parent(s):
4ac4212
bug pipeline
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -11,7 +11,7 @@ import operator
|
|
11 |
|
12 |
|
13 |
######## PredTitrage pipeline #########
|
14 |
-
class
|
15 |
|
16 |
def __init__(self, beam_size=5, batch_size=32, **kwargs):
|
17 |
self.beam_size = beam_size
|
|
|
11 |
|
12 |
|
13 |
######## PredTitrage pipeline #########
|
14 |
+
class PredTitragesPipeline(Pipeline):
|
15 |
|
16 |
def __init__(self, beam_size=5, batch_size=32, **kwargs):
|
17 |
self.beam_size = beam_size
|