include verb after name

app.py

@@ -11,7 +11,7 @@ def hello(prompt, items):
         inp += '.'
     inp += ' %'
     input_ids = torch.tensor([tokenizer.encode(inp)])
-    output = model.generate(input_ids, max_length=30)
+    output = model.generate(input_ids, max_length=35)
     resp = tokenizer.decode(output[0], skip_special_tokens=True)
     if '%' in resp:
         resp1 = resp[resp.index('%') + 1 : ]
@@ -30,8 +30,17 @@ def hello(prompt, items):
 
     if (names[0] in resp1) and ((names[1] not in resp1) or (resp1.index(names[0]) < resp1.index(names[1]))):
         force_inp = resp1[:resp1.index(names[0])] + names[1]
-    else:
+        remainder = resp1[resp1.index(names[0]) + len(names[0]):].strip().split(' ')
+    elif (names[1] in resp1):
         force_inp = resp1[:resp1.index(names[1])] + names[0]
+        remainder = resp1[resp1.index(names[1]) + len(names[1]):].strip().split(' ')
+    else:
+        return [resp1, 'Name not present']
+    if len(remainder) > 0:
+        if remainder[0] in ['is', 'are', 'was', 'were']:
+            force_inp += ' ' + ' '.join(remainder[:2])
+        else:
+            force_inp += ' ' + remainder[0]
     alt = inp + ' ' + force_inp
     input_ids2 = torch.tensor([tokenizer.encode(alt)])
     output2 = model.generate(input_ids2, max_new_tokens=12, do_sample=True)