Spaces:
Runtime error
Runtime error
Pawel_Mar
commited on
Commit
·
f30955e
1
Parent(s):
dafdb47
working prototype (returns array casted to string)
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ def yoloV8_func(image: gr.Image = None):
|
|
| 6 |
model_path = "models/last.pt"
|
| 7 |
model = YOLO(model_path)
|
| 8 |
results = model.predict(image)
|
| 9 |
-
mask_json = results[0].masks.
|
| 10 |
|
| 11 |
return str(mask_json)
|
| 12 |
|
|
|
|
| 6 |
model_path = "models/last.pt"
|
| 7 |
model = YOLO(model_path)
|
| 8 |
results = model.predict(image)
|
| 9 |
+
mask_json = results[0].masks.xy
|
| 10 |
|
| 11 |
return str(mask_json)
|
| 12 |
|