a bit debug

modeling_ocrqa.py

@@ -84,14 +84,16 @@ class QAAssessmentModel(PreTrainedModel):
                         "other": self.config.config.filename["other"]}
 
         self.ocrqa_assessors = {}
+
         # model_filename = self.config.config.model[lang]
-        for lang in bin_filenames:
+        for lang in bin_filenames.keys():
             model_filename = self.config.config.filename[lang]
         # Check if the file is already present locally, else download it
             if not os.path.exists(model_filename):
                 # print(f"{bin_filename} not found locally, downloading from Hugging Face hub...")
                 self.ocrqa_assessors[lang] = hf_hub_download(repo_id=self.config.config._name_or_path,
                                                              filename=model_filename)
+        print(self.ocrqa_assessors)
         self.lang_pipeline = pipeline("langident",
                                     model="impresso-project/impresso-langident",
                                     trust_remote_code=True,