update

classification/tokenizer_bbb.py

@@ -39,10 +39,10 @@ class BBBTokenizer(PreTrainedTokenizer):
       )
     ])
 
-    model_dir = snapshot_download("SaeedLab/TITAN-BBB/classification", allow_patterns=["normalize"])
-    transformer_tab_path = os.path.join(model_dir, "normalize_tabular.joblib")
-    transformer_img_path = os.path.join(model_dir, "normalize_image.joblib")
-    transformer_txt_path = os.path.join(model_dir, "normalize_text.joblib")
+    model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_cls"])
+    transformer_tab_path = os.path.join(model_dir, "normalize_cls_tabular.joblib")
+    transformer_img_path = os.path.join(model_dir, "normalize_cls_image.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)

normalize_cls_image.joblib

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

normalize_cls_tabular.joblib

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

normalize_cls_text.joblib

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