Skip to content

Commit d2d2d9b

Browse files
committed
Update classifier.py
update modelcard generation
1 parent 3aff6d1 commit d2d2d9b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/adaptive_classifier/classifier.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -482,7 +482,7 @@ def _generate_model_card(self) -> str:
482482
from adaptive_classifier import AdaptiveClassifier
483483
484484
# Load the model
485-
classifier = AdaptiveClassifier.from_pretrained("{self.model.config._name_or_path}")
485+
classifier = AdaptiveClassifier.from_pretrained("adaptive-classifier/model-name")
486486
487487
# Make predictions
488488
text = "Your text here"

0 commit comments

Comments
 (0)