Jiahuita
/

NewsSourceClassification

Text Classification

news-classification

Model card Files Files and versions Community

Jiahuita commited on 16 days ago

Commit

1833f3a

•

1 Parent(s): bb8ddb0

same issue diff place

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -26,7 +26,7 @@ class NewsClassifierConfig(PretrainedConfig):
 class NewsClassifier(PreTrainedModel):
     config_class = NewsClassifierConfig
-    base_model_prefix = "news_classifier"
     def __init__(self, config):
         super().__init__(config)

 class NewsClassifier(PreTrainedModel):
     config_class = NewsClassifierConfig
+    base_model_prefix = "text_classifier"
     def __init__(self, config):
         super().__init__(config)