Spaces:
Running
Running
Commit
·
845a442
1
Parent(s):
1f21dba
use generic model
Browse files- service.py +2 -2
service.py
CHANGED
@@ -13,8 +13,8 @@ class PretrainedModelRunnable(bentoml.Runnable):
|
|
13 |
SUPPORTS_CPU_MULTI_THREADING = True
|
14 |
|
15 |
def __init__(self):
|
16 |
-
self.unmasker = pipeline('object-detection'
|
17 |
-
|
18 |
|
19 |
@bentoml.Runnable.method(batchable=False)
|
20 |
def __call__(self, input_text):
|
|
|
13 |
SUPPORTS_CPU_MULTI_THREADING = True
|
14 |
|
15 |
def __init__(self):
|
16 |
+
self.unmasker = pipeline('object-detection') #model="davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks",
|
17 |
+
# feature_extractor="davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks")
|
18 |
|
19 |
@bentoml.Runnable.method(batchable=False)
|
20 |
def __call__(self, input_text):
|