reach-vb
/

musicgen-small-endpoint

Inference Endpoints

Model card Files Files and versions Community

Vaibhav Srivastav commited on Jul 27, 2023

Commit

6388023

•

1 Parent(s): 6d3baac

up

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -20,7 +20,7 @@ class EndpointHandler:
         # preprocess
         inputs = self.processor(
-            text=inputs,
             padding=True,
             return_tensors="pt",)

         # preprocess
         inputs = self.processor(
+            text=[inputs],
             padding=True,
             return_tensors="pt",)