impresso-project
/

ocr-quality-assessor-unigram-light

Token Classification

ocr_qa_assessment

quality-assessment

Model card Files Files and versions Community

emanuelaboros commited on Apr 15

Commit

777cb0a

·

1 Parent(s): ba1cf57

a bit debug

Files changed (1) hide show

modeling_ocrqa.py +1 -1

modeling_ocrqa.py CHANGED Viewed

@@ -112,7 +112,7 @@ class QAAssessmentModel(PreTrainedModel):
             # [{'label': 'fr', 'confidence': 99.87}]
             if len(langs) > 0:
                 print(f"Detected languages: {langs}")
-                lang = langs[0]['label']
                 logger.info(f"Detected language: {lang}")
             else:
                 lang = "other"

             # [{'label': 'fr', 'confidence': 99.87}]
             if len(langs) > 0:
                 print(f"Detected languages: {langs}")
+                lang = langs['language']
                 logger.info(f"Detected language: {lang}")
             else:
                 lang = "other"