segmentation-backend

Sleeping

diegokauer commited on Dec 27, 2023

Commit

40bdaa3

1 Parent(s): aa96f98

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -44,8 +44,8 @@ class Model(LabelStudioMLBase):
     def __init__(self):
-    self.image_processor = AutoImageProcessor.from_pretrained("diegokauer/conditional-detr-coe-int")
-    self.model = AutoModelForObjectDetection.from_pretrained("diegokauer/conditional-detr-coe-int")
     def predict(self, tasks, **kwargs):
         """ This is where inference happens: model returns

     def __init__(self):
+        self.image_processor = AutoImageProcessor.from_pretrained("diegokauer/conditional-detr-coe-int")
+        self.model = AutoModelForObjectDetection.from_pretrained("diegokauer/conditional-detr-coe-int")
     def predict(self, tasks, **kwargs):
         """ This is where inference happens: model returns