Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,6 +11,6 @@ if img is not None:
|
|
| 11 |
results = model(img)
|
| 12 |
st.image(img,'Input')
|
| 13 |
for result in results:
|
| 14 |
-
for
|
| 15 |
mask = mask.numpy()* 255
|
| 16 |
st.image(mask,'Mask')
|
|
|
|
| 11 |
results = model(img)
|
| 12 |
st.image(img,'Input')
|
| 13 |
for result in results:
|
| 14 |
+
for mask in result.masks.data:
|
| 15 |
mask = mask.numpy()* 255
|
| 16 |
st.image(mask,'Mask')
|