Commit
·
ba1cf57
1
Parent(s):
b828aa0
a bit debug
Browse files- modeling_ocrqa.py +1 -0
modeling_ocrqa.py
CHANGED
|
@@ -111,6 +111,7 @@ class QAAssessmentModel(PreTrainedModel):
|
|
| 111 |
langs = self.lang_pipeline(input_ids)
|
| 112 |
# [{'label': 'fr', 'confidence': 99.87}]
|
| 113 |
if len(langs) > 0:
|
|
|
|
| 114 |
lang = langs[0]['label']
|
| 115 |
logger.info(f"Detected language: {lang}")
|
| 116 |
else:
|
|
|
|
| 111 |
langs = self.lang_pipeline(input_ids)
|
| 112 |
# [{'label': 'fr', 'confidence': 99.87}]
|
| 113 |
if len(langs) > 0:
|
| 114 |
+
print(f"Detected languages: {langs}")
|
| 115 |
lang = langs[0]['label']
|
| 116 |
logger.info(f"Detected language: {lang}")
|
| 117 |
else:
|