haukurpj commited on
Commit
d7ae5c8
·
1 Parent(s): 37e6c2b

Add readme with directions on how to run the model

Browse files
Files changed (1) hide show
  1. README.md +36 -0
README.md ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - is
4
+ tags:
5
+ - nlp
6
+ - pos
7
+ library_name: transformers
8
+ paper: https://arxiv.org/abs/2201.05601
9
+ ---
10
+ ```python
11
+ from transformers import AutoModel, AutoTokenizer
12
+
13
+ model = AutoModel.from_pretrained("mideind/IceBERT-PoS", trust_remote_code=True)
14
+ tokenizer = AutoTokenizer.from_pretrained("mideind/IceBERT-PoS")
15
+
16
+ # Pre tokenized
17
+ sentence = "Ég veit að þú kemur í kvöld til mín ."
18
+
19
+ result = model.predict_labels_from_text([sentence], tokenizer)
20
+ expected = [
21
+ [
22
+ ("fp", ["1", "sing", "nom"]),
23
+ ("sf", ["sing", "act", "1", "pres"]),
24
+ ("c", []),
25
+ ("fp", ["2", "sing", "nom"]),
26
+ ("sf", ["sing", "act", "2", "pres"]),
27
+ ("af", ["pos"]),
28
+ ("n", ["neut", "sing", "acc"]),
29
+ ("af", ["pos"]),
30
+ ("fp", ["1", "sing", "gen"]),
31
+ ("pl", []),
32
+ ]
33
+ ]
34
+ assert result == expected, f"Expected {expected}, but got {result}"
35
+ print("Test passed successfully!")
36
+ ```