anegda commited on
Commit
f9e99f2
·
verified ·
1 Parent(s): 900d68a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +42 -3
README.md CHANGED
@@ -1,3 +1,42 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ - es
6
+ - eu
7
+ metrics:
8
+ - BLEU
9
+ - TER
10
+ ---
11
+ ## Medical English-Basque machine translation model
12
+
13
+ ## Model description
14
+
15
+ - **Model type:** translation
16
+ - **Source Language:** English / Spanish
17
+ - **Target Language:** Basque
18
+ - **License:** apache-2.0
19
+
20
+ ## How to Get Started with the Model
21
+
22
+ Use the code below to get started with the model.
23
+
24
+ ```
25
+ from transformers import MarianMTModel, MarianTokenizer
26
+ from transformers import AutoTokenizer
27
+ from transformers import AutoModelForSeq2SeqLM
28
+
29
+ src_text = ["The patient had brain damage", "El paciente tiene daño cerebral"]
30
+
31
+ model_name = "anegda/medical_enes-eu"
32
+ tokenizer = MarianTokenizer.from_pretrained(model_name)
33
+
34
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
35
+ translated = model.generate(**tokenizer(src_text, return_tensors="pt", padding=True))
36
+ print([tokenizer.decode(t, skip_special_tokens=True) for t in translated])`
37
+ ```
38
+
39
+ The recommended environments include the following transfomer versions: 4.12.3 , 4.15.0 , 4.26.1
40
+
41
+ ### Contact information
42
+ For further information, send an email to <[email protected]>