fix: handler.py

handler.py

@@ -2,11 +2,12 @@ from typing import Dict, List, Any
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
 
-class PhiHandler():
+class EndpointHandler():
     def __init__(self, path=""):
         self.model = AutoModelForCausalLM.from_pretrained(path, device_map="cuda", torch_dtype="auto", trust_remote_code=True)
         self.tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct")
         self.pipe = pipeline("text-generation", model=self.model, tokenizer=self.tokenizer)
+
     def __call__(self, data:Dict[str, Any]) :
         messages = data.pop("messages", None)
         generation_args = data.pop("generation_args", None)