Update pipeline.py

pipeline.py

@@ -88,7 +88,8 @@ class MOSDiffusionPipeline(DiffusionPipeline):
         Build dataset_key dynamically from the provided prompt.
         """
         # for simplicity let's just return the prompt as the dataset_key
-        return {"prompt": prompt}
+        data = [{"wav": "", "caption": prompt}]  # no wav file, just the caption (prompt)
+        return {"data": data}
         
 
 # Example of how to use the pipeline