Skip to content

Commit

Permalink
fix docker
Browse files Browse the repository at this point in the history
  • Loading branch information
danielgatis committed Jun 30, 2023
1 parent a9026d0 commit acc526f
Showing 1 changed file with 1 addition and 7 deletions.
8 changes: 1 addition & 7 deletions rembg/sessions/u2net_custom.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,13 +23,7 @@ def __init__(
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(
Expand Down

0 comments on commit acc526f

Please sign in to comment.