remove load ckpt and img from model

app.py

@@ -9,13 +9,13 @@ from cell_segmentation.inference.inference_cellvit_experiment_monuseg import Inf
 
 ## local |  remote
 RUN_MODE = "remote"
-if RUN_MODE != "local":
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/model_best.pth")
-    ## examples
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/1.png")
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/2.png")
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/3.png")
-    os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/4.png")
+# if RUN_MODE != "local":
+#     os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/model_best.pth")
+#     ## examples
+#     os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/1.png")
+#     os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/2.png")
+#     os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/3.png")
+#     os.system("wget https://huggingface.co/xiazhi/LKCell-demo/resolve/main/4.png")
 
 ## step 1: set up model
 

cell_segmentation/inference/inference_cellvit_experiment_monuseg.py

@@ -876,7 +876,6 @@ class MoNuSegInference:
         placeholder[:  h, 3 * w : 4 * w, :3] = np.asarray(pred_cell_image) / 255
 
         # plotting
-        test_image = Image.fromarray((placeholder * 255).astype(np.uint8))
         fig, axs = plt.subplots(figsize=(3, 2), dpi=1200)
         axs.imshow(placeholder)
         axs.set_xticks(np.arange(w / 2, 4 * w, w))