2
0
Daniel Gatis 2 жил өмнө
parent
commit
acc526fbaf

+ 1 - 7
rembg/sessions/u2net_custom.py

@@ -23,13 +23,7 @@ class U2netCustomSession(BaseSession):
         if model_path is None:
         if model_path is None:
             raise ValueError("model_path is required")
             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]:
     def predict(self, img: PILImage, *args, **kwargs) -> List[PILImage]:
         ort_outs = self.inner_session.run(
         ort_outs = self.inner_session.run(