metadata
license: apache-2.0
metrics:
- comet
Creator: Unbabel
The Hub was created to enable the direct usage of the wmt21-comet-qe-mqm model with Python from the Hub.
Code example:
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)