wmt21-comet-qe-mqm / README.md
zwhe99's picture
Update README.md
bdd1e56
---
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)
```