Spaces:
Sleeping
Sleeping
diegokauer
commited on
Commit
•
a06c3a6
1
Parent(s):
40bdaa3
Update model.py
Browse files
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 |
-
|
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
|