ft_handler fix, remove unrequired indexing (deepjavalibrary#736)

Lokiiiiii · May 22, 2023 · 3d733b0 · 3d733b0
1 parent eaa5394
commit 3d733b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/engines/python/setup/djl_python/fastertransformer.py b/engines/python/setup/djl_python/fastertransformer.py
@@ -115,7 +115,7 @@ def inference(self, inputs: Input):
  batch_size=len(input_text),
  output_len=output_len,
  beam_width=beam_width,
- **parameters)[0]
+ **parameters)
  result = [{"generated_text": s} for s in result]
  outputs = Output().add(result)
  except Exception as e: