diegokauer commited on
Commit
a06c3a6
1 Parent(s): 40bdaa3

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +2 -5
model.py CHANGED
@@ -41,11 +41,8 @@ def generate_download_signed_url_v4(blob_name):
41
  class Model(LabelStudioMLBase):
42
 
43
  os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = get_credentials()
44
-
45
- def __init__(self):
46
-
47
- self.image_processor = AutoImageProcessor.from_pretrained("diegokauer/conditional-detr-coe-int")
48
- self.model = AutoModelForObjectDetection.from_pretrained("diegokauer/conditional-detr-coe-int")
49
 
50
  def predict(self, tasks, **kwargs):
51
  """ This is where inference happens: model returns
 
41
  class Model(LabelStudioMLBase):
42
 
43
  os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = get_credentials()
44
+ image_processor = AutoImageProcessor.from_pretrained("diegokauer/conditional-detr-coe-int")
45
+ model = AutoModelForObjectDetection.from_pretrained("diegokauer/conditional-detr-coe-int")
 
 
 
46
 
47
  def predict(self, tasks, **kwargs):
48
  """ This is where inference happens: model returns