From acc526fbafd0b2043cdaa07d15491be4ffa55428 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Thu, 29 Jun 2023 21:20:45 -0300 Subject: [PATCH] fix docker --- rembg/sessions/u2net_custom.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) 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(