emanuelaboros commited on
Commit
88c1917
·
1 Parent(s): 27d03da

a bit debug

Browse files
Files changed (1) hide show
  1. 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
- # Check if the file is already present locally, else download it
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,