added model url to default params

handler.py

@@ -3,7 +3,7 @@ from transformers import AutoProcessor, MusicgenForConditionalGeneration
 import torch
 
 class EndpointHandler:
- def __init__(self, path=""):
+ def __init__(self, path="pbotsaris/musicgen-small"):
  # load model and processor
  self.processor = AutoProcessor.from_pretrained(path)
  self.model = MusicgenForConditionalGeneration.from_pretrained(path, torch_dtype=torch.float16)
@@ -36,3 +36,7 @@ class EndpointHandler:
 
  return [{"audio": pred, "sr": self.model.config.sampling_rate}]
 
+
+
+if __name__ == "__main__":
+ handler = EndpointHandler()