diff --git a/Cellpose/CellposePrediction.ipynb b/Cellpose/CellposePrediction.ipynb
index 5f6f337..f94bab5 100644
--- a/Cellpose/CellposePrediction.ipynb
+++ b/Cellpose/CellposePrediction.ipynb
@@ -46,7 +46,7 @@
     "    elif option == 'cellpose':\n",
     "        zip_file_url = 'https://github.com/ajinkya-kulkarni/TrainedModels/raw/main/Cellpose/cellpose_model.zip'\n",
     "        zip_filename = 'CellposeTrainedModel.zip'\n",
-    "        model_folder_path = 'cellpose_model.941139'\n",
+    "        model_folder_path = 'cellpose_model.693768'\n",
     "    else:\n",
     "        raise ValueError(f\"Invalid option: {option}\")\n",
     "\n",
@@ -100,7 +100,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "918f2fb1fc1c40bba470f9f237043fd8",
+       "model_id": "47c7efaef3124d21b89245e1ec0fd0ad",
        "version_major": 2,
        "version_minor": 0
       },
@@ -114,7 +114,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "c6eba8a748be40d09cf503640e3b7c9a",
+       "model_id": "1cdf1abc4dc54f36b8bfd8c5ce44caff",
        "version_major": 2,
        "version_minor": 0
       },
@@ -220,7 +220,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "29c32c4c0ef8430d8b5fd17da8970367",
+       "model_id": "519572741c504e28b09f3ece55272b67",
        "version_major": 2,
        "version_minor": 0
       },
diff --git a/Cellpose/CellposeTraining.py b/Cellpose/CellposeTraining.py
index 9339a14..2818cc4 100644
--- a/Cellpose/CellposeTraining.py
+++ b/Cellpose/CellposeTraining.py
@@ -85,7 +85,7 @@ def copy_files(src_root, filenames, destination):
 min_labels_per_mask = 2
 initial_diameter = 0
 pretrained_model = 'None'
-batch_size = 2
+batch_size = 4
 flow_threshold = 0.4
 base_dir = "/home/ajinkya/Desktop/PyOrganoidAnalysis/Cellpose"
 GPU_USAGE = True
@@ -137,4 +137,4 @@ def copy_files(src_root, filenames, destination):
 print('Training model successful.')
 print()
 
-##################################################################################################################################################
\ No newline at end of file
+##################################################################################################################################################