Upload utils.py

src/utils.py

@@ -9,12 +9,11 @@ def post_process_template(tB):
 
 
 def construct_template(words, templateA, if_then=False):
-    if len(words) == 2:
-        # template = ['{} <mask> {}.'.format(words[0], words[1])]
-        templates = [
-            # '{} is <mask> {}.'.format(words[0], words[1]), 
-            '{} <mask> {}.'.format(words[0], words[1]),
-        ]
+    if len(words) >= 2:
+        templates = ['{} <mask> '.format(words[0])]
+        for i in range(1, len(words)-1):
+            templates[0] += '{} <mask> '.format(words[i])
+        templates[0] += '{}.'.format(words[-1])
     elif len(words) == 1:
         templates = [
             # '{} is <mask>.'.format(words[0]),