flaviagiammarino
/

medsam-vit-base

Mask Generation

Inference Endpoints

Model card Files Files and versions Community

flaviagiammarino commited on Jul 12, 2023

Commit

6a5851a

•

1 Parent(s): 5622644

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -28,7 +28,7 @@ import requests
 import numpy as np
 import matplotlib.pyplot as plt
 from PIL import Image
-from transformers import SamModel, SamProcessor, SamImageProcessor
 import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -36,7 +36,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 model = SamModel.from_pretrained("flaviagiammarino/medsam-vit-base").to(device)
 processor = SamProcessor.from_pretrained("flaviagiammarino/medsam-vit-base")
-img_url = "https://raw.githubusercontent.com/bowang-lab/MedSAM/main/assets/img_demo.png"
 raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")
 input_boxes = [95., 255., 190., 350.]

 import numpy as np
 import matplotlib.pyplot as plt
 from PIL import Image
+from transformers import SamModel, SamProcessor
 import torch
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model = SamModel.from_pretrained("flaviagiammarino/medsam-vit-base").to(device)
 processor = SamProcessor.from_pretrained("flaviagiammarino/medsam-vit-base")
+img_url = "https://huggingface.co/flaviagiammarino/medsam-vit-base/resolve/main/scripts/input.png"
 raw_image = Image.open(requests.get(img_url, stream=True).raw).convert("RGB")
 input_boxes = [95., 255., 190., 350.]