Update model.py

model.py

@@ -69,7 +69,7 @@ class Model(LabelStudioMLBase):
 
             result_list = []
             for score, label, box in zip(results['scores'], results['labels'], results['boxes']):
-                label_id = str(uuid4())[:4]
+                label_id = str(uuid4())
                 x, y, x2, y2 = tuple(box)
                 result_list.append({
                     'id': label_id,
@@ -80,17 +80,17 @@ class Model(LabelStudioMLBase):
                     'type': 'labels',
                     'score': score.item(),  # per-region score, visible in the editor 
                     'value': {
-                        'x': x,
-                        'y': y,
-                        'width': x2-x,
-                        'height': y2-y,
+                        'x': x.item(),
+                        'y': y.item(),
+                        'width': (x2-x).item(),
+                        'height': (y2-y).item(),
                         'rotation': 0,
                         'labels': [self.id2label[label.item()]]
                     }
                 })
             
             predictions.append({
-                'score': results['scores'].mean(),  # prediction overall score, visible in the data manager columns
+                'score': results['scores'].mean().item(),  # prediction overall score, visible in the data manager columns
                 'model_version': 'diegokauer/conditional-detr-coe-int',  # all predictions will be differentiated by model version
                 'result': result_list
             })