Update handler.py

handler.py

@@ -15,13 +15,13 @@ class EndpointHandler:
     def __init__(self, path=""):
         device = "cuda" if torch.cuda.is_available() else "cpu"
         config = XttsConfig()
-        config.load_json("./model/config.json")
+        config.load_json("/repository/model/config.json")
         model = Xtts.init_from_config(config)
         model.load_checkpoint(
             config,
-            checkpoint_path="./model/model.pth",
-            vocab_path="./model/vocab.json",
-            speaker_file_path="./model/speakers_xtts.pth",
+            checkpoint_path="/repository/model/model.pth",
+            vocab_path="/repository/model/vocab.json",
+            speaker_file_path="/repository/model/speakers_xtts.pth",
             eval=True,
             use_deepspeed=device == "cuda",
         )
@@ -35,7 +35,7 @@ class EndpointHandler:
             gpt_cond_latent,
             speaker_embedding,
         ) = self.model.get_conditioning_latents(
-            audio_path="attenborough.mp3",
+            audio_path="/repository/attenborough.mp3",
             gpt_cond_len=30,
             gpt_cond_chunk_len=4,
             max_ref_length=60,