Update pipeline.py

pipeline.py

@@ -4,14 +4,15 @@ from transformers import SamModel, SamProcessor
 from PIL import Image
 import numpy as np
 
+MODEL_ID = "facebook/sam-vit-huge"
+
 class PreTrainedPipeline():
     def __init__(self, path=""):
 
         self.device = torch.device(
             "cuda" if torch.cuda.is_available() else "cpu")
-        self.processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
-        self.model = SamModel.from_pretrained(
-            "facebook/sam-vit-base").to(self.device)
+        self.processor = SamProcessor.from_pretrained(MODEL_ID)
+        self.model = SamModel.from_pretrained(MODEL_ID).to(self.device)
         self.model.eval()
         self.model = self.model.to(self.device)