drab
/

distribution-equipment-belgium

Object Detection

Model card Files Files and versions Community

drab commited on Apr 4, 2022

Commit

d2812b8

·

1 Parent(s): 2e69dbe

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -14,7 +14,7 @@ from PIL import Image
 class PreTrainedPipeline():
     def __init__(self, path: str):
         # load the model
-        self.model = tf.saved_model.load()
     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:

 class PreTrainedPipeline():
     def __init__(self, path: str):
         # load the model
+        self.model = tf.saved_model.load(os.path.join(path, "saved_model.pb"))
     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]: