Update README.md
Browse files
README.md
CHANGED
@@ -9,4 +9,17 @@ pipeline_tag: text-classification
|
|
9 |
|
10 |
# greek-court-decision-paragraph-classifier
|
11 |
|
12 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
|
10 |
# greek-court-decision-paragraph-classifier
|
11 |
|
12 |
+
```python
|
13 |
+
model_path = hf_hub_download(repo_id="pprokopidis/greek-admin-court-decisions-paragraph-classifier", filename="20241124-model.ftz")
|
14 |
+
sample_decision = hf_hub_download(repo_id="pprokopidis/greek-admin-court-decisions-paragraph-classifier", filename="sample_data/Α2485_2023.txt")
|
15 |
+
model = load_model(model_path)
|
16 |
+
labels_map = {'preamble': '__label__0', '__label__0': 'preamble', 'panel': '__label__1', '__label__1': 'panel', 'litigants': '__label__2', '__label__2': 'litigants', 'justification': '__label__3', '__label__3': 'justification', 'decision': '__label__4', '__label__4': 'decision', 'post': '__label__5', '__label__5': 'post'}
|
17 |
+
with open(sample_decision) as inf:
|
18 |
+
text = NL.join([p for p in inf.read().split(NL) if p.strip()])
|
19 |
+
nchars = 150
|
20 |
+
for line in text.split(NL):
|
21 |
+
pred = labels_map[model.predict(line.strip())[0][0]]
|
22 |
+
if len(line) > nchars:
|
23 |
+
line = line[0:nchars]
|
24 |
+
print(f"{line} -> {pred}")
|
25 |
+
```
|