Update README.md

README.md

@@ -17,6 +17,8 @@ pipeline_tag: image-segmentation
 from huggingface_hub import from_pretrained_keras
 
 model = from_pretrained_keras("Plsek/CADET-v1")
+
+y_pred = model.predict(X.reshape(1,128,128,1))
 ```
 
 ## How to cite