SaeedLab
/

TITAN-BBB

blood-brain-barrier

Model card Files Files and versions

gabrielbianchin commited on 22 days ago

Commit

7af20d3

·

1 Parent(s): f5400b9

update

Files changed (1) hide show

classification/modeling_bbb.py +1 -1

classification/modeling_bbb.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import PreTrainedModel
 from transformers.modeling_outputs import BaseModelOutputWithPooling, SequenceClassifierOutput
 import os
-from classification.configuration_bbb import BBBConfig
 class BBBModelForSequenceClassification(PreTrainedModel):
   config_class = BBBConfig

 from transformers.modeling_outputs import BaseModelOutputWithPooling, SequenceClassifierOutput
 import os
+from .classification.configuration_bbb import BBBConfig
 class BBBModelForSequenceClassification(PreTrainedModel):
   config_class = BBBConfig