davanstrien HF Staff commited on
Commit
845a442
·
1 Parent(s): 1f21dba

use generic model

Browse files
Files changed (1) hide show
  1. 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', 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):
 
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):