davanstrien HF staff commited on
Commit
f78ab1e
1 Parent(s): 905a09b

Update service.py

Browse files
Files changed (1) hide show
  1. 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="davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks",
12
- feature_extractor="davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks")
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):