def predict()

in src/autofill_model.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

    def predict(self, html_tags: list[str]):
        if self.pipe is None:
            self.pipe = pipeline('text-classification', model=self.save_path, truncation=True, batch_size=8)
        return self.pipe(html_tags)