Commit ·
378285d
1
Parent(s): 7b6c2c3
Update README.md
Browse files
README.md
CHANGED
|
@@ -14,13 +14,14 @@ More information needed
|
|
| 14 |
|
| 15 |
Use the code below to get started with the model.
|
| 16 |
|
| 17 |
-
|
|
|
|
| 18 |
notebook_login()
|
| 19 |
|
| 20 |
from huggingface_hub import from_pretrained_keras
|
| 21 |
|
| 22 |
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)
|
| 23 |
-
|
| 24 |
|
| 25 |
|
| 26 |
## Training and evaluation data
|
|
|
|
| 14 |
|
| 15 |
Use the code below to get started with the model.
|
| 16 |
|
| 17 |
+
```
|
| 18 |
+
from huggingface_hub import notebook_login
|
| 19 |
notebook_login()
|
| 20 |
|
| 21 |
from huggingface_hub import from_pretrained_keras
|
| 22 |
|
| 23 |
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)
|
| 24 |
+
```
|
| 25 |
|
| 26 |
|
| 27 |
## Training and evaluation data
|