drab commited on
Commit
134b360
1 Parent(s): d2812b8

Simple debug version

Browse files
pipeline.py CHANGED
@@ -14,7 +14,7 @@ from PIL import Image
14
  class PreTrainedPipeline():
15
  def __init__(self, path: str):
16
  # load the model
17
- self.model = tf.saved_model.load(os.path.join(path, "saved_model.pb"))
18
 
19
  def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:
20
 
 
14
  class PreTrainedPipeline():
15
  def __init__(self, path: str):
16
  # load the model
17
+ self.model = tf.saved_model.load(os.path.join(path, "saved_model"))
18
 
19
  def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:
20
 
saved_model.pb → saved_model/saved_model.pb RENAMED
File without changes