Update handler.py

handler.py

@@ -12,7 +12,7 @@ nltk.download("averaged_perceptron_tagger_eng")
 NEL_MODEL = "nel-mgenre-multilingual"
 
 class NelPipeline:
-    def __init__(self, model_name: str):
+    def __init__(self, model_dir: str = "."):
         self.model_name = model_name
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -149,7 +149,7 @@ def get_wikipedia_title(qid, language="en"):
 class EndpointHandler:
     def __init__(self, path: str = None):
         # Initialize the NelPipeline with the specified model
-        self.pipeline = NelPipeline(NEL_MODEL)
+        self.pipeline = NelPipeline()
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         # Process incoming data