emanuelaboros commited on
Commit
fce343d
·
1 Parent(s): cdd5fd4

add mainr eadme

Browse files
Files changed (2) hide show
  1. README.md +5 -6
  2. 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 = "emanuelaboros/lang-detect"
21
 
22
- lang_pipeline = pipeline("lang-detect", model=MODEL_NAME,
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
- langs = lang_pipeline(sentence)
30
- langs
31
 
32
  ```
33
 
34
  ```
35
- {'label': 'fr', 'confidence': 99.87}
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
- "lang-detect": {
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
  }