update

normalize_reg_image.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0ccbeff5f432bdacd79818161a1be46a1284e79cdabf1d0336dfaed667b6eca9
+size 49767

normalize_reg_tabular.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f9924bf96594b1a75996aaef81b95a948c0406e66511702d08065e8e1ad4e2e1
+size 9831

normalize_reg_text.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6f95a0d363340f0e3d05cd6c2ab861dd8a2d047363bcc3f9f31dd13d6f236c1a
+size 19047

regression/tokenizer_bbb.py

@@ -39,12 +39,16 @@ class BBBTokenizer(PreTrainedTokenizer):
       )
     ])
 
-    #model_dir = snapshot_download("SaeedLab/BBBP-Classification", allow_patterns=["transformer.pkl"])
-    #transformer_path = os.path.join(model_dir, "transformer.pkl")
-
-    self.feature_transformer_tab = joblib.load('classification/normalize_tabular.joblib')
-    self.feature_transformer_img = joblib.load('classification/normalize_image.joblib')
-    self.feature_transformer_txt = joblib.load('classification/normalize_text.joblib')
+    model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_tabular.joblib"])
+    transformer_tab_path = os.path.join(model_dir, "normalize_cls_tabular.joblib")
+    model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_image.joblib"])
+    transformer_img_path = os.path.join(model_dir, "normalize_cls_image.joblib")
+    model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_reg_text.joblib"])
+    transformer_txt_path = os.path.join(model_dir, "normalize_cls_text.joblib")
+
+    self.feature_transformer_tab = joblib.load(transformer_tab_path)
+    self.feature_transformer_img = joblib.load(transformer_img_path)
+    self.feature_transformer_txt = joblib.load(transformer_txt_path)
 
   def generate_tab_features(self, smiles):
     mol = Chem.MolFromSmiles(smiles)