Update app.py

app.py

@@ -410,9 +410,9 @@ with gr.Blocks(
     gr.Markdown("# CellposeCellCounter")
     gr.Markdown("For accurate cell confluency, crop the image to display only desired area.")
         
-    # Define State components to store masks and image data across function calls
-    masks_state = pack_array(masks)
-    image_state = pack_array(processed_image_np)
+    masks_state = gr.State(value=None)
+    image_state = gr.State(value=None)
+
 
     with gr.Tab("Image Editor (Draw Selection)"):
         gr.Markdown("### Draw selection and run segmentation")