use nls model

service.py

@@ -2,10 +2,6 @@ import bentoml
 
 from bentoml.io import Image, JSON
 from transformers import pipeline
-from transformers import AutoFeatureExtractor, AutoModelForObjectDetection
-import transformers
-
-transformers.logging.set_verbosity_info()
 
 
 class PretrainedModelRunnable(bentoml.Runnable):
@@ -13,8 +9,8 @@ class PretrainedModelRunnable(bentoml.Runnable):
     SUPPORTS_CPU_MULTI_THREADING = True
 
     def __init__(self):
-        self.unmasker = pipeline('object-detection') #model="davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks",
-        #                          feature_extractor="davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks")
+        self.unmasker = pipeline('object-detection', model="davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks",
+                                  feature_extractor="davanstrien/detr-resnet-50_fine_tuned_nls_chapbooks")
 
     @bentoml.Runnable.method(batchable=False)
     def __call__(self, input_text):