Simple debug version

Files changed (2) hide show

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(os.path.join(path, "saved_model.pb"))
     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"))
     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:

saved_model.pb → saved_model/saved_model.pb RENAMED Viewed

File without changes