fix docker

danielgatis · Jun 30, 2023 · a9026d0 · a9026d0
1 parent b527af6
commit a9026d0
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/rembg/sessions/u2net_custom.py b/rembg/sessions/u2net_custom.py
@@ -2,6 +2,7 @@
 from typing import List
 
 import numpy as np
+import onnxruntime as ort
 import pooch
 from PIL import Image
 from PIL.Image import Image as PILImage