|
--- |
|
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) |
|
``` |