SaeedLab
/

TITAN-BBB

blood-brain-barrier

Model card Files Files and versions

gabrielbianchin commited on about 1 month ago

Commit

dc76faa

·

1 Parent(s): 482b2e5

update

Files changed (1) hide show

classification/tokenizer_bbb.py +1 -1

classification/tokenizer_bbb.py CHANGED Viewed

@@ -39,7 +39,7 @@ class BBBTokenizer(PreTrainedTokenizer):
       )
     ])
-    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")

       )
     ])
+    model_dir = snapshot_download("SaeedLab/TITAN-BBB", allow_patterns=["normalize_cls_tabular.joblib"])
     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")