id2label

id2label.py

 print(f"Running inference on {config.s3_parent_dir}/run_{config.run_num}")

 data_parent_path = f"{config.s3_parent_dir}/data/prepared_data/{config.encoded_data_dir}"
  id2label_local_path = os.path.join(os.getcwd(), 'id2label.json')
  if not os.path.isfile(id2label_local_path):
      id2label_s3_path = f"{data_parent_path}/id2label.json"
      bucket.download_file(id2label_s3_path, id2label_local_path)
      with open(id2label_local_path) as f:
          id2label = json.load(f)