update chexbert

CheXbert/src/label.py

@@ -9,7 +9,8 @@ from weights_utils import get_weight
 
 # sys.path.append(os.path.abspath('/home/gholipos-admin/Desktop/Thesis/Training_Code/Entity_Summarization/CheXbert/src'))
 
-import utils
+from . import utils
+# import utils
 from models.bert_labeler import bert_labeler
 from bert_tokenizer import tokenize
 from transformers import BertTokenizer

CheXbert/src/utils.py

@@ -4,7 +4,8 @@ import torch.nn as nn
 import pandas as pd
 import numpy as np
 import json
-from models.bert_labeler import bert_labeler
+# from models.bert_labeler import bert_labeler
+from .models.bert_labeler import bert_labeler
 from bert_tokenizer import tokenize
 from sklearn.metrics import f1_score, confusion_matrix
 from statsmodels.stats.inter_rater import cohens_kappa

inference.py

@@ -13,13 +13,13 @@ import sys, os
 # ---------------------------------------------------------------------
 # Make CheXbert's `src` folder importable (so `import utils` works)
 # ---------------------------------------------------------------------
-BASE_DIR = os.path.dirname(__file__)
-CHEXBERT_SRC = os.path.join(BASE_DIR, "CheXbert", "src")
+# BASE_DIR = os.path.dirname(__file__)
+# CHEXBERT_SRC = os.path.join(BASE_DIR, "CheXbert", "src")
 
-if CHEXBERT_SRC not in sys.path:
-    sys.path.insert(0, CHEXBERT_SRC)
+# if CHEXBERT_SRC not in sys.path:
+#     sys.path.insert(0, CHEXBERT_SRC)
 
-from label import label  # now imports /app/CheXbert/src/label.py
+# from label import label  # now imports /app/CheXbert/src/label.py
 import pandas as pd
 import numpy as np
 import time
@@ -116,7 +116,7 @@ from DETR.arguments import get_args_parser as get_detr_args_parser
 from VG import localization
 from ssim import ssim
 
-# from CheXbert.src.label import label
+from CheXbert.src.label import label
 
 def get_args_parser():
     parser = argparse.ArgumentParser('Set the Input', add_help=True)