Spaces:
Sleeping
Sleeping
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
|