Spaces:
Runtime error
Runtime error
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):
|