jens commited on
Commit
f465c1d
·
1 Parent(s): 344f858
Files changed (2) hide show
  1. app.py +1 -1
  2. inference.py +1 -0
app.py CHANGED
@@ -20,7 +20,7 @@ def snap(image, video):
20
 
21
  demo = gr.Interface(
22
  snap,
23
- inputs=[gr.Image(source="webcam", tool=None),
24
  gr.Video(source="webcam")],
25
  outputs=[gr.Image(label="RGB"),
26
  gr.Image(label="predicted depth"),
 
20
 
21
  demo = gr.Interface(
22
  snap,
23
+ inputs=[gr.Image(source="webcam", tool=None, label="Input Image", type="pil"),
24
  gr.Video(source="webcam")],
25
  outputs=[gr.Image(label="RGB"),
26
  gr.Image(label="predicted depth"),
inference.py CHANGED
@@ -47,6 +47,7 @@ class SegmentPredictor:
47
  self.mask_generator = SamAutomaticMaskGenerator(sam)
48
 
49
  def predict(self, image):
 
50
  sam_result = self.mask_generator.generate(image)
51
  mask_annotator = sv.MaskAnnotator()
52
  detections = sv.Detections.from_sam(sam_result=sam_result)
 
47
  self.mask_generator = SamAutomaticMaskGenerator(sam)
48
 
49
  def predict(self, image):
50
+ image = np.array(image)
51
  sam_result = self.mask_generator.generate(image)
52
  mask_annotator = sv.MaskAnnotator()
53
  detections = sv.Detections.from_sam(sam_result=sam_result)