nateraw commited on
Commit
ad7e72a
1 Parent(s): 6f55e39

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +5 -3
pipeline.py CHANGED
@@ -1,5 +1,7 @@
1
  class Pipeline:
2
- def __init__(self, model_id):
3
- self.model_id = model_id
 
 
4
  def __call__(self, inputs):
5
- return f"Hello, {inputs.title()}"
1
  class Pipeline:
2
+ def __init__(self, prefix: str = "Good morning,", postfix: str = '!'):
3
+ self.prefix = prefix
4
+ self.postfix = postfix
5
+
6
  def __call__(self, inputs):
7
+ return f"{self.prefix} {inputs.title()}{self.postfix}"