Upload app.py with huggingface_hub

app.py

@@ -54,11 +54,18 @@ def generate(req: GenerateRequest):
         with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as f:
             tmp_path = f.name
 
+        # Use voice reference if it exists
+        voice_ref = Path("/app/voice-reference.wav")
+        audio_prompt = str(voice_ref) if voice_ref.exists() else None
+        if audio_prompt:
+            print(f"[dia2] Using voice clone from {audio_prompt}", flush=True)
+
         start = time.time()
         result = dia.generate(
             req.text,
             config=config,
             output_wav=tmp_path,
+            audio_prompt=audio_prompt,
             verbose=True,
         )
         elapsed = time.time() - start