We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 74fda8b commit 1ef9e4fCopy full SHA for 1ef9e4f
ppstructure/kie/predict_kie_token_ser.py
@@ -115,7 +115,9 @@ def __call__(self, img):
115
else:
116
data[idx] = [data[idx]]
117
if self.args.use_onnx:
118
- input_tensor = {name: data[idx] for idx, name in enumerate(self.input_tensor)}
+ input_tensor = {
119
+ name: data[idx] for idx, name in enumerate(self.input_tensor)
120
+ }
121
self.output_tensors = self.predictor.run(None, input_tensor)
122
123
for idx in range(len(self.input_tensor)):
@@ -125,7 +127,9 @@ def __call__(self, img):
125
127
126
128
outputs = []
129
for output_tensor in self.output_tensors:
- output = output_tensor if self.args.use_onnx else output_tensor.copy_to_cpu()
130
+ output = (
131
+ output_tensor if self.args.use_onnx else output_tensor.copy_to_cpu()
132
+ )
133
outputs.append(output)
134
preds = outputs[0]
135
0 commit comments