Update service.py

service.py

@@ -8,12 +8,12 @@ class PretrainedModelRunnable(bentoml.Runnable):
     SUPPORTS_CPU_MULTI_THREADING = True
 
     def __init__(self):
-        self.unmasker = pipeline('object-detection', model="biglam/detr-resnet-50_fine_tuned_nls_chapbooks",
+        self.object_detector = pipeline('object-detection', model="biglam/detr-resnet-50_fine_tuned_nls_chapbooks",
                                   feature_extractor="biglam/detr-resnet-50_fine_tuned_nls_chapbooks")
 
-    @bentoml.Runnable.method(batchable=True)
-    def __call__(self, input_text):
-        return self.unmasker(input_text)
+    @bentoml.Runnable.method(batchable=False)
+    def __call__(self, input_image):
+        return self.object_detector(input_image)
 
 
 runner = bentoml.Runner(PretrainedModelRunnable, name="pretrained_illustration_detection")