Spaces:
Running
Running
Commit
·
f78ab1e
1
Parent(s):
905a09b
Update service.py
Browse files- service.py +2 -2
service.py
CHANGED
|
@@ -8,8 +8,8 @@ class PretrainedModelRunnable(bentoml.Runnable):
|
|
| 8 |
SUPPORTS_CPU_MULTI_THREADING = True
|
| 9 |
|
| 10 |
def __init__(self):
|
| 11 |
-
self.unmasker = pipeline('object-detection', model="
|
| 12 |
-
feature_extractor="
|
| 13 |
|
| 14 |
@bentoml.Runnable.method(batchable=True)
|
| 15 |
def __call__(self, input_text):
|
|
|
|
| 8 |
SUPPORTS_CPU_MULTI_THREADING = True
|
| 9 |
|
| 10 |
def __init__(self):
|
| 11 |
+
self.unmasker = pipeline('object-detection', model="biglam/detr-resnet-50_fine_tuned_nls_chapbooks",
|
| 12 |
+
feature_extractor="biglam/detr-resnet-50_fine_tuned_nls_chapbooks")
|
| 13 |
|
| 14 |
@bentoml.Runnable.method(batchable=True)
|
| 15 |
def __call__(self, input_text):
|