Update app.py

app.py

@@ -70,6 +70,21 @@ async def generate(prompt: str, duration: float = 10.0):
 
     audio = output.audios[0]
 
+    # Convert torch → numpy
+    if isinstance(audio, torch.Tensor):
+        audio = audio.detach().cpu().numpy()
+    
+    # If shape is (channels, samples), transpose it
+    if audio.ndim == 2 and audio.shape[0] < audio.shape[1]:
+        audio = audio.T
+    
+    # Ensure float32
+    audio = audio.astype("float32")
+    
+    # Clamp to valid range
+    audio = audio.clip(-1.0, 1.0)
+
+
     filename = f"{uuid.uuid4().hex}.wav"
     path = os.path.join(OUTPUT_DIR, filename)