added create_params func

handler.py

@@ -5,6 +5,33 @@ import torch
 import io
 import base64
 
+def create_params(params, fr):
+
+    # default
+    out = { "do_sample": True,
+          "guidance_scale": 3, 
+          "max_new_tokens": 256
+          }
+
+    has_tokens = False
+
+    if params is None:
+       return out
+
+    if 'duration' in params:
+        out['max_new_tokens'] =  params['duration'] * fr
+        has_tokens = True
+
+    for k, p in params.items():
+        if k in out: 
+          if has_tokens and k == 'max_new_tokens':
+            continue
+
+          out[k] = p
+
+    return out
+
+
 class EndpointHandler:
     def __init__(self, path="pbotsaris/musicgen-small"):
         self.processor = AutoProcessor.from_pretrained(path)