keras-io
/

semantic-segmentation

Image Segmentation

Model card Files Files and versions Community

merve HF staff commited on Nov 26, 2021

Commit

aadf252

·

1 Parent(s): 9dd290a

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -14,7 +14,7 @@ from PIL import Image
 class PreTrainedPipeline():
     def __init__(self, model_id: str):
-        self.model = keras.models.load_model(model_id)
     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:

 class PreTrainedPipeline():
     def __init__(self, model_id: str):
+        self.model = keras.models.load_model(./tf_model.h5)
     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]: