Commit
·
88c1917
1
Parent(s):
27d03da
a bit debug
Browse files- modeling_ocrqa.py +1 -1
modeling_ocrqa.py
CHANGED
@@ -88,7 +88,7 @@ class QAAssessmentModel(PreTrainedModel):
|
|
88 |
# model_filename = self.config.config.model[lang]
|
89 |
for lang in bin_filenames.keys():
|
90 |
model_filename = self.config.config.filename[lang]
|
91 |
-
|
92 |
if not os.path.exists(model_filename):
|
93 |
# print(f"{bin_filename} not found locally, downloading from Hugging Face hub...")
|
94 |
self.ocrqa_assessors[lang] = hf_hub_download(repo_id=self.config.config._name_or_path,
|
|
|
88 |
# model_filename = self.config.config.model[lang]
|
89 |
for lang in bin_filenames.keys():
|
90 |
model_filename = self.config.config.filename[lang]
|
91 |
+
print(f"Loading model for {lang}: {model_filename}")
|
92 |
if not os.path.exists(model_filename):
|
93 |
# print(f"{bin_filename} not found locally, downloading from Hugging Face hub...")
|
94 |
self.ocrqa_assessors[lang] = hf_hub_download(repo_id=self.config.config._name_or_path,
|