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

repaired import

Browse files
Files changed (2) hide show
  1. modeling_ocrqa.py +1 -1
  2. test.py +14 -0
modeling_ocrqa.py CHANGED
@@ -5,7 +5,7 @@ import logging
5
  import floret
6
  import os
7
  from huggingface_hub import hf_hub_download
8
- from .configuration_lang import ImpressoConfig
9
 
10
  logger = logging.getLogger(__name__)
11
  from pybloomfilter import BloomFilter
 
5
  import floret
6
  import os
7
  from huggingface_hub import hf_hub_download
8
+ from .configuration_ocrqa import ImpressoConfig
9
 
10
  logger = logging.getLogger(__name__)
11
  from pybloomfilter import BloomFilter
test.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ MODEL_NAME = "impresso-project/ocr-quality-assessment-light"
4
+
5
+ ocrqa_pipeline = pipeline("ocr-qa-assessment", model=MODEL_NAME,
6
+ trust_remote_code=True,
7
+ device='cpu')
8
+
9
+ sentence = """En l'an 1348, au plus fort des ravages de la peste noire à travers l'Europe,
10
+ le Royaume de France se trouvait à la fois au bord du désespoir et face à une opportunité."""
11
+
12
+ score = ocrqa_pipeline(sentence)
13
+ print(score)
14
+