Upload Colony_Analyzer_AI2_HF.py

Colony_Analyzer_AI2_HF.py

@@ -40,10 +40,12 @@ def pad(img_np, tw=2048, th=1536):
 #this is the huggingface version
 def cut_img(img, patch_size=512):
     img_map = {}
-    if hasattr(img, "size"):
-        width, height = img.size  # PIL Image
+    if isinstance(img, Image.Image):
+        width, height = img.size
+        img_np = np.array(img)
     else:
-        height, width = img.shape[:2]  # NumPy array
+        img_np = img
+        height, width = img_np.shape[:2]
     i_num = height // patch_size
     j_num = width // patch_size
     count = 1