Update model.py

model.py

@@ -41,8 +41,8 @@ def generate_download_signed_url_v4(blob_name):
 class Model(LabelStudioMLBase):
     
     os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = get_credentials()
-    image_processor = AutoImageProcessor.from_pretrained("diegokauer/conditional-detr-coe-int")
-    model = AutoModelForObjectDetection.from_pretrained("diegokauer/conditional-detr-coe-int")
+    image_processor = AutoImageProcessor.from_pretrained("diegokauer/conditional-detr-coe-int-v2")
+    model = AutoModelForObjectDetection.from_pretrained("diegokauer/conditional-detr-coe-int-v2")
     seg_image_processor = ViTImageProcessor.from_pretrained("diegokauer/int-pet-classifier-v2")
     seg_model = Swinv2ForImageClassification.from_pretrained("diegokauer/int-pet-classifier-v2")
     id2label = model.config.id2label
@@ -132,7 +132,7 @@ class Model(LabelStudioMLBase):
             
             predictions.append({
                 'score': results['scores'].mean().item(),  # prediction overall score, visible in the data manager columns
-                'model_version': 'cdetr_v2',  # all predictions will be differentiated by model version
+                'model_version': 'cdetr_v2.5',  # all predictions will be differentiated by model version
                 'result': result_list
             })
         print(predictions)