Add error handling for model downloads and reorder imports in session files

rembg/bg.py

@@ -210,7 +210,10 @@ def download_models(models: tuple[str, ...]) -> None:
             sys.exit(1)
         else:
             print(f"Downloading model: {model}")
-            session.download_models()
+            try:
+                session.download_models()
+            except Exception as e:
+                print(f"Error downloading model: {e}")
 
 
 def remove(

rembg/sessions/ben_custom.py

@@ -1,16 +1,13 @@
 import os
 from typing import List
-import onnxruntime as ort
-import numpy as np
 
+import numpy as np
+import onnxruntime as ort
 from PIL import Image
 from PIL.Image import Image as PILImage
 
 from .base import BaseSession
 
-import numpy as np
-from PIL import Image
-
 
 class BenCustomSession(BaseSession):
     """This is a class representing a custom session for the Ben model."""

rembg/sessions/dis_custom.py

@@ -1,8 +1,8 @@
 import os
 from typing import List
-import onnxruntime as ort
-import numpy as np
 
+import numpy as np
+import onnxruntime as ort
 from PIL import Image
 from PIL.Image import Image as PILImage