Update model.py

model.py

@@ -43,8 +43,8 @@ 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")
-    seg_image_processor = ViTImageProcessor.from_pretrained("diegokauer/int-pet-classifier")
-    seg_model = Swinv2ForImageClassification.from_pretrained("diegokauer/int-pet-classifier")
+    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
     seg_id2label = seg_model.config.id2label