Update app.py

app.py

@@ -33,9 +33,9 @@ def get_audio_from_yt_video(yt_link: str):
 
 def inference(file_uploaded_in, file_uploaded_ref):
 
-    output_wav = None
+    output_wav_path = None
     
-    return output_wav
+    return output_wav_path, output_wav_path
 
 
 with gr.Blocks() as demo:
@@ -105,13 +105,15 @@ with gr.Blocks() as demo:
             <div> <h3> <center> Mixing Style Transferred Output. </h3> </div>
             """
         )
-        # with gr.Row().style(mobile_collapse=False, equal_height=True):
+        with gr.Row().style(mobile_collapse=False, equal_height=True):
+            output_mix_l = gr.Audio(label="Listen to output mix")
+            output_mix_p = gr.Audio(label="Download output mix")
         with gr.Row():
             output_mix = gr.File(label="Download style transferred music track")
             inference_btn.click(
                 inference,
                 inputs=[file_uploaded_in, file_uploaded_ref],
-                outputs=[wav_output],
+                outputs=[output_mix_l, output_mix_p],
             )