Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import gradio as gr
 from dia.model import Dia
 from huggingface_hub import InferenceClient
+import numpy as np
 
 # Hardcoded podcast subject
 PODCAST_SUBJECT = "The future of AI and its impact on society"
@@ -44,13 +45,14 @@ def split_podcast_into_chunks(podcast_text, chunk_size=3):
 
 def process_audio_chunks(podcast_text):
     chunks = split_podcast_into_chunks(podcast_text)
-
+    sample_rate = 36000 # Modified from https://huggingface.co/spaces/nari-labs/Dia-1.6B/blob/main/app.py has 44100
     for chunk in chunks:
         if stop_signal.is_set():
             break
 
-        audio_chunk = model.generate(chunk, use_torch_compile=True, verbose=False)
-        audio_queue.put(audio_chunk)
+        raw_audio = model.generate(chunk, use_torch_compile=True, verbose=False)
+        audio_chunk = np.array(raw_audio, dtype=np.float32)
+        audio_queue.put((sample_rate, audio_chunk))
 
     audio_queue.put(None)
 
@@ -63,8 +65,6 @@ def stream_audio_generator(podcast_text):
     gen_thread = threading.Thread(target=process_audio_chunks, args=(podcast_text,))
     gen_thread.start()
 
-    sample_rate = 32000 # Modified from https://huggingface.co/spaces/nari-labs/Dia-1.6B/blob/main/app.py has 44100
-
     try:
         while True:
             # Get next chunk from queue
@@ -75,7 +75,7 @@ def stream_audio_generator(podcast_text):
                 break
 
             # Yield the audio chunk with sample rate
-            yield (sample_rate, chunk)
+            yield chunk
 
     except Exception as e:
         print(f"Error in streaming: {e}")