AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-18 08:45:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix docker

Browse Source
This commit is contained in:
Daniel Gatis 2023-06-29 21:20:45 -03:00
parent a9026d02e0
commit acc526fbaf
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
rembg/sessions/u2net_custom.py
View File

@ -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__( super().__init__(model_name, sess_opts, providers, *args, **kwargs)
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(