diegokauer commited on
Commit
858a727
·
1 Parent(s): f5aeda9

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -107,7 +107,7 @@ class Model(LabelStudioMLBase):
107
 
108
  predictions.append({
109
  'score': results['scores'].mean().item(), # prediction overall score, visible in the data manager columns
110
- 'model_version': 'diegokauer/conditional-detr-coe-int', # all predictions will be differentiated by model version
111
  'result': result_list
112
  })
113
  print(predictions)
 
107
 
108
  predictions.append({
109
  'score': results['scores'].mean().item(), # prediction overall score, visible in the data manager columns
110
+ 'model_version': 'cdetr_v1', # all predictions will be differentiated by model version
111
  'result': result_list
112
  })
113
  print(predictions)