Merge pull request #107 from alexey-krasnov/master

fix: remove strict string check to support pathlib objects in remove_transparent function
Kohulan · Sep 9, 2024 · fa3f097 · fa3f097
2 parents e55c377 + 4a1c71b
commit fa3f097
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/DECIMER/config.py b/DECIMER/config.py
@@ -153,12 +153,11 @@ def handle_numpy_array(array: np.ndarray) -> Image.Image:
         png = Image.fromarray(array).convert("RGBA")
         return process_image(png)
 
-    if isinstance(image, str):
-        return handle_image_path(image_path=image)
-    elif isinstance(image, np.ndarray):
+    # Check if input is a numpy array
+    if isinstance(image, np.ndarray):
         return handle_numpy_array(array=image)
-    else:
-        raise ValueError("Input should be either a string (image path) or a numpy array.")
+
+    return handle_image_path(image_path=image)
 
 
 def get_bnw_image(image):