fix zonos enhance order

Author: painebenjamin

setup.py

@@ -59,7 +59,7 @@
 
 setup(
     name="taproot",
-    version="0.3.6",  # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
+    version="0.3.7",  # expected format is one of x.y.z.dev0, or x.y.z.rc1 or x.y.z (no to dashes, yes to dots)
     description="Taproot is a seamlessly scalable AI/ML inference engine designed for deployment across hardware clusters with disparate capabilities.",
     long_description=open("README.md", "r", encoding="utf-8").read(),
     long_description_content_type="text/markdown",

src/taproot/tasks/generation/audio/speech/zonos/task.py

@@ -259,6 +259,7 @@ def synthesize(
 
             if enhance:
                 audio, _ = self.enhance(audio, self.sample_rate)
+                audio = audio.unsqueeze(0)
 
             if (i < num_texts - 1):
                 if "speaker" in unconditional_keys: