diff --git a/rembg/sessions/u2net_custom.py b/rembg/sessions/u2net_custom.py index 2350a8e..58cd383 100644 --- a/rembg/sessions/u2net_custom.py +++ b/rembg/sessions/u2net_custom.py @@ -23,13 +23,7 @@ class U2netCustomSession(BaseSession): if model_path is None: raise ValueError("model_path is required") - super().__init__( - model_name, - sess_opts, - providers, - *args, - **kwargs - ) + super().__init__(model_name, sess_opts, providers, *args, **kwargs) def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]: ort_outs = self.inner_session.run(