Update pipeline.py

pipeline.py

@@ -6,7 +6,7 @@ import numpy as np
 import json
 
 class NewsClassifierConfig(PretrainedConfig):
-    model_type = "news_classifier"
+    model_type = "custom"
     
     def __init__(
         self,
@@ -26,7 +26,7 @@ class NewsClassifierConfig(PretrainedConfig):
 
 class NewsClassifier(PreTrainedModel):
     config_class = NewsClassifierConfig
-    base_model_prefix = "news_classifier"
+    base_model_prefix = "custom"
 
     def __init__(self, config):
         super().__init__(config)
@@ -35,7 +35,7 @@ class NewsClassifier(PreTrainedModel):
         
     def post_init(self):
         """Load model and tokenizer after initialization"""
-        self.model = load_model('news_classifier.h5')
+        self.model = load_model('custom.h5')
         with open('tokenizer.json', 'r') as f:
             tokenizer_data = json.load(f)
             self.tokenizer = tokenizer_from_json(tokenizer_data)