Update app.py

app.py

@@ -37,7 +37,7 @@ def inference(image, model_path, **model_params):
     img = image
 
     model_inference = models.CellposeModel(gpu=False, pretrained_model=model_path)
-    preds, flows, _ = model_inference.eval(img, **model_params)
+    preds, flows, _ = model_inference.eval([img], **model_params)
 
     print(preds.shape)
     return preds, flows
@@ -48,11 +48,11 @@ if __name__ == "__main__":
     st.title("Sartorius Cell Segmentation")
 
     uploaded_img = st.file_uploader(label="Upload neuronal cell image")
+    st.image(uploaded_img)
     segment = st.button("Perform segmentation")
     if uploaded_img is not None and segment:
         img = Image.open(uploaded_img)
         img = np.array(img)
-        st.image(img)
 
         model_params = {
             "diameter": 19.0,
@@ -60,10 +60,11 @@ if __name__ == "__main__":
             "augment": True,
             "resample": True,
         }
-        preds, flows = inference(
-            image=img,
-            model_path="cellpose_residual_on_style_on_concatenation_off_fold1_ep_649_cv_0.2834",
-            **model_params
-        )
+        with st.spinner("Performing segmentation. This might take a while..."):
+            preds, flows = inference(
+                image=img,
+                model_path="./cellpose_residual_on_style_on_concatenation_off_fold1_ep_649_cv_0.2834",
+                **model_params
+            )
 
         print(preds)