fix: append

andreiionutdamian · Apr 22, 2024 · bf2c9ac · bf2c9ac
1 parent 45024da
commit bf2c9ac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/src/serving/serving_app.py b/app/src/serving/serving_app.py
@@ -323,8 +323,8 @@ def _predict_job (self, model_type: str, input, params:dict = None):
               if isinstance(input, List):
                 batch = []
                 for cnt in range(batch_size):
-                  if (tndx*batch_size+cnt < len(input)):
-                    batch[cnt] = input[tndx*batch_size+cnt]
+                  if ((tndx*batch_size+cnt) < len(input)):
+                    batch.append(input[tndx*batch_size+cnt])
 
                 if isinstance(batch[0], str):
                   content_list.append(json.dumps(batch))