Update handler.py

handler.py

@@ -5,7 +5,7 @@ import torch
 class EndpointHandler:
     def __init__(self, model_path):
         self.processor = AutoProcessor.from_pretrained(model_path)
-        self.model = MusicgenForConditionalGeneration.from_pretrained(model_path)
+        self.model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
         if torch.cuda.is_available():
             self.model = self.model.to("cuda")