Update src/api/ModelMethods.py

src/api/ModelMethods.py

@@ -78,15 +78,16 @@ def inputSentenceList(input):
 def processSentenceList(model, inputSentenceList):
     out_sentence = []
     for s in inputSentenceList:
-        if model == "mT5":
-            out_s = mt5Inference(s)
-        elif model == "mBART":
-            out_s = mbartInference(s)
-        elif model == "VartaT5":
-            out_s = vartat5Inference(s)
-        else:
-            return f"Model: {model} not available"
-        out_sentence.append(out_s[0]["sequence"])
+        if(len(s)>2):
+            if model == "mT5":
+                out_s = mt5Inference(s)
+            elif model == "mBART":
+                out_s = mbartInference(s)
+            elif model == "VartaT5":
+                out_s = vartat5Inference(s)
+            else:
+                return f"Model: {model} not available"
+            out_sentence.append(out_s[0]["sequence"])
     return out_sentence