File size: 921 Bytes
6562c05 bdd1e56 6562c05 bdd1e56 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
---
license: apache-2.0
metrics:
- comet
---
Creator: [Unbabel](https://unbabel.github.io/COMET/html/index.html)
The Hub was created to enable the direct usage of the wmt21-comet-qe-mqm model with Python from the Hub.
Code example:
```python
pip install --upgrade pip # ensures that pip is current
pip install unbabel-comet
from comet import download_model, load_from_checkpoint
model_path = download_model("zwhe99/wmt21-comet-qe-mqm")
model = load_from_checkpoint(model_path)
data = [
{
"src": "Dem Feuer konnte Einhalt geboten werden",
"mt": "The fire could be stopped",
"ref": "They were able to control the fire."
},
{
"src": "Schulen und Kindergärten wurden eröffnet.",
"mt": "Schools and kindergartens were open",
"ref": "Schools and kindergartens opened"
}
]
model_output = model.predict(data, batch_size=8, gpus=1)
print (model_output)
``` |