File size: 919 Bytes
fc126bc
 
fd1ed12
 
fc126bc
5d53f9f
fd1ed12
5d53f9f
642ca66
5d53f9f
fd1ed12
5d53f9f
fd1ed12
5d53f9f
 
 
 
 
fd1ed12
5d53f9f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
fd1ed12
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-da 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-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)
```