segmentation-backend

Sleeping

diegokauer commited on Dec 27, 2023

Commit

4f251ee

1 Parent(s): 49221d6

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -78,7 +78,7 @@ class Model(LabelStudioMLBase):
                     'from_name': "label",
                     'to_name': "image",
                     'type': 'labels',
-                    'score': score,  # per-region score, visible in the editor
                     'value': {
                         'x': x,
                         'y': y,

                     'from_name': "label",
                     'to_name': "image",
                     'type': 'labels',
+                    'score': score.item(),  # per-region score, visible in the editor
                     'value': {
                         'x': x,
                         'y': y,