소스 검색

fix docker

Daniel Gatis 2 년 전
부모
커밋
a9026d02e0
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      rembg/sessions/u2net_custom.py

+ 1 - 0
rembg/sessions/u2net_custom.py

@@ -2,6 +2,7 @@ import os
 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