Daniele Licari commited on
Commit
0039f4f
·
1 Parent(s): 3c27e83

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +13 -0
README.md CHANGED
@@ -28,4 +28,17 @@ from transformers import AutoModel, AutoTokenizer
28
  model_name = "dlicari/Italian-Legal-BERT"
29
  tokenizer = AutoTokenizer.from_pretrained(model_name)
30
  model = AutoModel.from_pretrained(model_name)
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  ```
 
28
  model_name = "dlicari/Italian-Legal-BERT"
29
  tokenizer = AutoTokenizer.from_pretrained(model_name)
30
  model = AutoModel.from_pretrained(model_name)
31
+ ```
32
+
33
+ You can use the Transformers library fill-mask pipeline to do inference with ITALIAN-LEGAL-BERT.
34
+ ```python
35
+ from transformers import pipeline
36
+ model_name = "dlicari/Italian-Legal-BERT"
37
+ fill_mask = pipeline("fill-mask", model_name)
38
+ fill_mask("Il [MASK] ha chiesto revocarsi l'obbligo di pagamento")
39
+ #[{'sequence': "Il ricorrente ha chiesto revocarsi l'obbligo di pagamento",'score': 0.7264330387115479},
40
+ # {'sequence': "Il convenuto ha chiesto revocarsi l'obbligo di pagamento",'score': 0.09641049802303314},
41
+ # {'sequence': "Il resistente ha chiesto revocarsi l'obbligo di pagamento",'score': 0.039877112954854965},
42
+ # {'sequence': "Il lavoratore ha chiesto revocarsi l'obbligo di pagamento",'score': 0.028993653133511543},
43
+ # {'sequence': "Il Ministero ha chiesto revocarsi l'obbligo di pagamento", 'score': 0.025297977030277252}]
44
  ```