Commit
·
fce343d
1
Parent(s):
cdd5fd4
add mainr eadme
Browse files- README.md +5 -6
- config.json +1 -1
README.md
CHANGED
@@ -11,28 +11,27 @@ tags:
|
|
11 |
|
12 |
#### How to use
|
13 |
|
14 |
-
You can use this model with Transformers *pipeline* for NER.
|
15 |
|
16 |
<!-- Provide a longer summary of what this model is. -->
|
17 |
```python
|
18 |
from transformers import pipeline
|
19 |
|
20 |
-
MODEL_NAME = "
|
21 |
|
22 |
-
|
23 |
trust_remote_code=True,
|
24 |
device='cpu')
|
25 |
|
26 |
sentence = """En l'an 1348, au plus fort des ravages de la peste noire à travers l'Europe,
|
27 |
le Royaume de France se trouvait à la fois au bord du désespoir et face à une opportunité."""
|
28 |
|
29 |
-
|
30 |
-
|
31 |
|
32 |
```
|
33 |
|
34 |
```
|
35 |
-
|
36 |
```
|
37 |
Works with lists of sentences also.
|
38 |
|
|
|
11 |
|
12 |
#### How to use
|
13 |
|
|
|
14 |
|
15 |
<!-- Provide a longer summary of what this model is. -->
|
16 |
```python
|
17 |
from transformers import pipeline
|
18 |
|
19 |
+
MODEL_NAME = "impresso-project/ocr-quality-assessment-light"
|
20 |
|
21 |
+
ocrqa_pipeline = pipeline("ocr-qa-assessment", model=MODEL_NAME,
|
22 |
trust_remote_code=True,
|
23 |
device='cpu')
|
24 |
|
25 |
sentence = """En l'an 1348, au plus fort des ravages de la peste noire à travers l'Europe,
|
26 |
le Royaume de France se trouvait à la fois au bord du désespoir et face à une opportunité."""
|
27 |
|
28 |
+
score = ocrqa_pipeline(sentence)
|
29 |
+
print(score)
|
30 |
|
31 |
```
|
32 |
|
33 |
```
|
34 |
+
|
35 |
```
|
36 |
Works with lists of sentences also.
|
37 |
|
config.json
CHANGED
@@ -15,7 +15,7 @@
|
|
15 |
"AutoModelForTokenClassification": "modeling_ocrqa.QAAssessmentModel"
|
16 |
},
|
17 |
"custom_pipelines": {
|
18 |
-
"
|
19 |
"impl": "ocr_qa_assessment.QAAssessmentPipeline",
|
20 |
"pt": "AutoModelForTokenClassification"
|
21 |
}
|
|
|
15 |
"AutoModelForTokenClassification": "modeling_ocrqa.QAAssessmentModel"
|
16 |
},
|
17 |
"custom_pipelines": {
|
18 |
+
"ocr-qa-assessment": {
|
19 |
"impl": "ocr_qa_assessment.QAAssessmentPipeline",
|
20 |
"pt": "AutoModelForTokenClassification"
|
21 |
}
|