Update README.md

README.md

@@ -10,6 +10,17 @@ More information needed
 
 More information needed
 
+## Use in Keras
+
+
+from huggingface_hub import notebook_login
+notebook_login()
+
+from huggingface_hub import from_pretrained_keras
+
+model = from_pretrained_keras("MIDSCapstoneTeam/ContrailSentinel", custom_objects={'dice_loss_plus_5focal_loss': total_loss, 'jaccard_coef': jaccard_coef, 'IOU score' : sm.metrics.IOUScore(threshold=0.9, name="IOU score"), 'Dice Coeficient' : sm.metrics.FScore(threshold=0.6, name="Dice Coeficient")}, compile=False)
+
+
 ## Training and evaluation data
 
 More information needed