keras-io
/

semantic-segmentation

Image Segmentation

Model card Files Files and versions

merve HF Staff commited on Nov 26, 2021

Commit

39a4527

·

1 Parent(s): aadf252

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(./tf_model.h5)
     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]]: