def predict()

in src/infer_location.py [0:0]

• 7 lines of code
• 1 McCabe index (conditional complexity)

    def predict(self, inputs):
        input_ids, attention_mask = inputs
        # Run inference
        outputs = self.ort_session.run(None, {
            "input_ids": input_ids,
            "attention_mask": attention_mask
        })
        return outputs[0]