Jiahuita
/

NewsSourceClassification

Text Classification

news-classification

Model card Files Files and versions Community

Jiahuita commited on 16 days ago

Commit

bb8ddb0

•

1 Parent(s): acdeca7

modified error in pipeline

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -6,7 +6,7 @@ import numpy as np
 import json
 class NewsClassifierConfig(PretrainedConfig):
-    model_type = "news_classifier"
     def __init__(
         self,

 import json
 class NewsClassifierConfig(PretrainedConfig):
+    model_type = "text_classifier"
     def __init__(
         self,