upstreamapp
/

sam-hf-endpoint

Inference Endpoints

Model card Files Files and versions Community

baconseason commited on Aug 16, 2023

Commit

cbd0404

·

1 Parent(s): 24ec34e

handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -7,7 +7,7 @@ from cv2 import imencode
 from base64 import b64encode
 class EndpointHandler():
-  def __init__(self):
     self.device = "cuda" if torch.cuda.is_available() else "cpu"
     self.generator = pipeline("mask-generation", model="facebook/sam-vit-large", device=self.device)

 from base64 import b64encode
 class EndpointHandler():
+  def __init__(self, path=""):
     self.device = "cuda" if torch.cuda.is_available() else "cpu"
     self.generator = pipeline("mask-generation", model="facebook/sam-vit-large", device=self.device)