Jiahuita
/

NewsSourceClassification

Text Classification

news-classification

Eval Results (legacy)

Model card Files Files and versions

Jiahuita commited on Nov 22, 2024

Commit

58554e5

·

1 Parent(s): b2de734

Pipeline change

Files changed (1) hide show

pipeline.py +2 -2

pipeline.py CHANGED Viewed

@@ -11,10 +11,10 @@ def load_tokenizer(tokenizer_path):
 class NewsClassificationPipeline(Pipeline):
     def __init__(self, model=None, tokenizer=None, **kwargs):
         super().__init__(**kwargs)
-        model_path = os.path.join(os.path.dirname(__file__), 'news_classifier.h5')
         self.model = tf.keras.models.load_model(model_path)
-        tokenizer_path = os.path.join(os.path.dirname(__file__), 'tokenizer.json')
         self.tokenizer_config = load_tokenizer(tokenizer_path)
     def __call__(self, texts, **kwargs):

 class NewsClassificationPipeline(Pipeline):
     def __init__(self, model=None, tokenizer=None, **kwargs):
         super().__init__(**kwargs)
+        model_path = os.path.join(os.path.dirname(__file__), './news_classifier.h5')
         self.model = tf.keras.models.load_model(model_path)
+        tokenizer_path = os.path.join(os.path.dirname(__file__), './tokenizer.json')
         self.tokenizer_config = load_tokenizer(tokenizer_path)
     def __call__(self, texts, **kwargs):