audiogen-medium-endpoint

jamesdon commited on Oct 24, 2023

Commit

7c9edea

1 Parent(s): c77789f

fix

Files changed (1) hide show

handler.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 class EndpointHandler:
     def __init__(self, path=""):
         # load model and processor from path
-        path = "facebook/audiogen-medium"
         self.processor = AutoProcessor.from_pretrained(path)
         self.model = MusicgenForConditionalGeneration.from_pretrained(path).to("cuda")

 class EndpointHandler:
     def __init__(self, path=""):
         # load model and processor from path
+        path = "jamesdon/audiogen-medium-endpoint"
         self.processor = AutoProcessor.from_pretrained(path)
         self.model = MusicgenForConditionalGeneration.from_pretrained(path).to("cuda")