Spaces:
Runtime error
Runtime error
jens
commited on
Commit
·
f465c1d
1
Parent(s):
344f858
fix
Browse files- app.py +1 -1
- 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)
|