Update app.py

app.py

@@ -66,9 +66,9 @@ def inference(file_uploaded_in, file_uploaded_ref):
     args = set_up()
     
     inference_style_transfer = Mixing_Style_Transfer_Inference(args)
-    output_wav_path = inference_style_transfer.inference(file_uploaded_in, file_uploaded_ref)
+    output_wav_path, fin_data_out_mix = inference_style_transfer.inference(file_uploaded_in, file_uploaded_ref)
     
-    return output_wav_path
+    return (44100, fin_data_out_mix)
 
 
 
@@ -102,7 +102,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             with gr.Blocks():
                 with gr.Tab("Input Music"):
-                    file_uploaded_in = gr.Audio(label="Input track (mix) to be mixing style transferred", type="filepath")
+                    file_uploaded_in = gr.Audio(label="Input track (mix) to be mixing style transferred")
                 with gr.Tab("YouTube url"):
                     with gr.Row():
                         yt_link_in = gr.Textbox(
@@ -127,7 +127,7 @@ with gr.Blocks() as demo:
                     )
             with gr.Blocks():
                 with gr.Tab("Reference Music"):
-                    file_uploaded_ref = gr.Audio(label="Reference track (mix) to copy mixing style", type="filepath")
+                    file_uploaded_ref = gr.Audio(label="Reference track (mix) to copy mixing style")
                 with gr.Tab("YouTube url"):
                     with gr.Row():
                         yt_link_ref = gr.Textbox(
@@ -160,7 +160,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             inference_btn = gr.Button("Run Mixing Style Transfer")
         with gr.Row():
-            output_mix = gr.Audio(label="mixing style transferred music track")
+            output_mix = gr.Audio(label="mixing style transferred music track", type='numpy')
             inference_btn.click(
                 inference,
                 inputs=[file_uploaded_in, file_uploaded_ref],
@@ -170,5 +170,4 @@ with gr.Blocks() as demo:
 
     
 if __name__ == "__main__":
-    demo.launch(debug=True)
-    
+    demo.launch(debug=True)