diegokauer commited on
Commit
8f40b79
·
1 Parent(s): fcb8bb5

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +7 -1
model.py CHANGED
@@ -59,4 +59,10 @@ class Model(LabelStudioMLBase):
59
  'model_version': 'diegokauer/conditional-detr-coe-int', # all predictions will be differentiated by model version
60
  'result': result_list
61
  })
62
- return predictions
 
 
 
 
 
 
 
59
  'model_version': 'diegokauer/conditional-detr-coe-int', # all predictions will be differentiated by model version
60
  'result': result_list
61
  })
62
+ return predictions
63
+
64
+ def fit(self, annotations, **kwargs):
65
+ """ This is where training happens: train your model given list of annotations,
66
+ then returns dict with created links and resources
67
+ """
68
+ return {'path/to/created/model': 'my/model.bin'}