change number of splits

app.py

@@ -106,7 +106,7 @@ def count_barnacles(image_raw, split_num, progress=gr.Progress()):
 
     print(cropped_image.shape)
 
-    split_num = 5
+    split_num = 3
 
     x_inc = int(cropped_image.shape[0]/split_num)
     y_inc = int(cropped_image.shape[1]/split_num)
@@ -127,8 +127,6 @@ def count_barnacles(image_raw, split_num, progress=gr.Progress()):
             # plt.figure()
             # plt.imshow(small_image)
             # plt.axis('on')
-            progress(0, desc=f"Encoding crop {r*split_num + c}/{split_num ** 2}")
-            mask_generator.predictor.set_image(small_image)
             progress(0, desc=f"Generating masks for crop {r*split_num + c}/{split_num ** 2}")
             masks = mask_generator.generate(small_image)
             num_masks = len(masks)
@@ -145,7 +143,7 @@ def count_barnacles(image_raw, split_num, progress=gr.Progress()):
     
     progress(0, desc="Generating Plot")
     # Create a figure with a size of 10 inches by 10 inches
-    fig = plt.figure(figsize=(10, 10))
+    fig = plt.figure(figsize=(100, 100))
 
     # Display the image using the imshow() function
     # plt.imshow(cropped_image)