drab
/

distribution-equipment-belgium

Object Detection

Model card Files Files and versions Community

drab commited on Apr 4, 2022

Commit

068c7ac

·

1 Parent(s): ec8ec72

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -15,7 +15,7 @@ from PIL import Image
 class PreTrainedPipeline():
     def __init__(self, path: str):
         # load the model
-        self.model = keras.models.load_model(os.path.join(path, "tf_model.h5"))
     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('./saved_model')
     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]: