fix: use config.yaml

handler.py

@@ -7,10 +7,8 @@ import torchaudio
 
 class EndpointHandler:
     def __init__(self, path=""):
-        if path != "config.yaml":
-          path = "hicustomer/pyannote-speaker-diarization"
         # load the model
-        self.pipeline = Pipeline.from_pretrained(path)
+        self.pipeline = Pipeline.from_pretrained("config.yaml")
 
     def __call__(self, data: Dict[str, bytes]) -> Dict[str, str]:
         """

handler_test.py

@@ -1,6 +1,6 @@
 from handler import EndpointHandler
 
-my_handler = EndpointHandler(path="config.yaml")
+my_handler = EndpointHandler()
 
 with open("./handler_test.wav", "rb") as file:
     bytes_data = file.read()