zwhe99/wmt21-comet-qe-da

Creator: Unbabel

The Hub was created to enable the direct usage of the wmt21-comet-qe-da 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-da")
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)