From a9026d02e09446c4ec997cd2489b2f867ae9dea4 Mon Sep 17 00:00:00 2001 From: Daniel Gatis Date: Thu, 29 Jun 2023 21:15:20 -0300 Subject: [PATCH] fix docker --- rembg/sessions/u2net_custom.py | 1 + 1 file changed, 1 insertion(+) diff --git a/rembg/sessions/u2net_custom.py b/rembg/sessions/u2net_custom.py index de6346f..2350a8e 100644 --- a/rembg/sessions/u2net_custom.py +++ b/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