drab commited on
Commit
5ae0a01
1 Parent(s): aafa424

Simple debug version

Browse files
.DS_Store ADDED
Binary file (6.15 kB). View file
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('./saved_model')
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('saved_model.pb')
18
 
19
  def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:
20
 
saved_model/saved_model.pb → saved_model.pb RENAMED
File without changes
saved_model/.DS_Store ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d65165279105ca6773180500688df4bdc69a2c7b771752f0a46ef120b7fd8ec3
3
+ size 6148