From 43078ef64649d5b5d3c624ed632e37a46eccb056 Mon Sep 17 00:00:00 2001
From: Gagan Bhatia <49101362+gagan3012@users.noreply.github.com>
Date: Mon, 10 May 2021 10:34:12 -0700
Subject: [PATCH] Update pipeline.py

---
 keytotext/pipeline.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/keytotext/pipeline.py b/keytotext/pipeline.py
index 4630517..9d2e3d9 100644
--- a/keytotext/pipeline.py
+++ b/keytotext/pipeline.py
@@ -7,7 +7,7 @@
     PreTrainedModel,
     PreTrainedTokenizer,
 )
-
+from .newmodels import NMPipeline
 
 class K2TPipeline:
     def __init__(
@@ -87,6 +87,12 @@ def _tokenize(
             "model": "gagan3012/k2t-base",
         },
     },
+    "mrm8488/t5-base-finetuned-common_gen":{
+        "impl": NMPipeline,
+        "default": {
+            "model": "mrm8488/t5-base-finetuned-common_gen",
+        },
+    }
 }