Simple debug version

pipeline.py

@@ -1,7 +1,8 @@
 import json
 from typing import Any, Dict, List
+
 import tensorflow as tf
-from tensorflow.keras.models import load_model
+from tensorflow import keras
 import base64
 import io
 import os
@@ -19,7 +20,8 @@ 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"))
+		self.model = keras.models.load_model(os.path.join(path, "tf_model.h5"))
+        #self.model = tf.saved_model.load(os.path.join(path, "saved_model"))
 
     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]: