Update model.py

model.py

@@ -71,6 +71,22 @@ class Model(LabelStudioMLBase):
             for score, label, box in zip(results['scores'], results['labels'], results['boxes']):
                 label_id = str(uuid4())
                 x, y, x2, y2 = tuple(box)
+                result_list.append({
+                    'id': label_id,
+                    'original_width': original_width,
+                    'original_height': original_height,
+                    'from_name': "bbox",
+                    'to_name': "image",
+                    'type': 'rectangle',
+                    'score': score.item(),  # per-region score, visible in the editor 
+                    'value': {
+                        'x': x.item(),
+                        'y': y.item(),
+                        'width': (x2-x).item(),
+                        'height': (y2-y).item(),
+                        'rotation': 0,
+                    }
+                })
                 result_list.append({
                     'id': label_id,
                     'original_width': original_width,