Replace original sam with medsam

Dockerfile

@@ -18,7 +18,7 @@ RUN python3.10 -m pip install --no-cache-dir --upgrade pip && \
 
 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install ffmpeg libsm6 libxext6  -y && rm -rf /var/lib/apt/lists
 
-RUN python3.10 -m pip install --no-cache-dir git+https://github.com/facebookresearch/segment-anything.git
+RUN python3.10 -m pip install --no-cache-dir git+https://github.com/bowang-lab/MedSAM.git
 
 WORKDIR /code
 

app.py

@@ -10,7 +10,7 @@ from PIL import Image
 from streamlit_drawable_canvas import st_canvas
 from utils import utils
 
-PREDICTOR_MODEL, AUTOMASK_MODEL = utils.get_model('checkpoint/sam_vit_b_01ec64.pth')
+PREDICTOR_MODEL, AUTOMASK_MODEL = utils.get_model('checkpoint/medsam_vit_b.pth')
 
 
 def process_box(predictor_model, show_mask, radius_width):

checkpoint/{sam_vit_b_01ec64.pth → medsam_vit_b.pth}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ec2df62732614e57411cdcf32a23ffdf28910380d03139ee0f4fcbe91eb8c912
-size 375042383
+oid sha256:bb845f3f42f585634ead00fad7728f44ddbe369a1e8c55b0df3cce5f6e1c1977
+size 375045993