fix param name

TTLURL · Feb 6, 2022 · 90cf72b · 90cf72b
1 parent f3c95cc
commit 90cf72b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ddddocr/__init__.py b/ddddocr/__init__.py
@@ -1587,11 +1587,11 @@ def classification(self, img):
         if not isinstance(img, (bytes, str, pathlib.PurePath, Image.Image)):
             raise TypeError("未知图片类型")
         if isinstance(img, bytes):
-            image = Image.open(io.BytesIO(img_bytes))
+            image = Image.open(io.BytesIO(img))
         elif isinstance(img, Image.Image):
             image = img.copy()
         elif isinstance(img, str):
-            image = base64_to_image(img_base64)
+            image = base64_to_image(img)
         else:
             assert isinstance(img, pathlib.PurePath)
             image = Image.open(img)