Upload Colony_Analyzer_AI2_HF.py

Colony_Analyzer_AI2_HF.py

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