Update app.py

app.py

@@ -19,7 +19,6 @@ import pytube as pt
 from pytube.exceptions import VideoUnavailable
 
 from inference.style_transfer import *
-from inference.mastering_transfer import *
 
 
 yt_video_dir = "./yt_dir/0"
@@ -65,14 +64,6 @@ def inference(file_uploaded_in, file_uploaded_ref):
     
     return output_wav_path
 
-def inference_mastering(file_uploaded_in, file_uploaded_ref):
-    # Perform music mastering style transfer
-    args = set_up()
-    
-    inference_mastering_style_transfer = Mastering_Style_Transfer_Inference(args)
-    output_wav_path_mastering = inference_mastering_style_transfer.inference(file_uploaded_in, file_uploaded_ref)
-    
-    return output_wav_path_mastering
 
 
 with gr.Blocks() as demo:
@@ -96,8 +87,9 @@ with gr.Blocks() as demo:
     gr.Markdown(
         """
         This page is a Hugging Face interactive demo of the paper ["Music Mixing Style Transfer: A Contrastive Learning Approach to Disentangle Audio Effects"](https://huggingface.co/papers/2211.02247) (ICASSP 2023).
-
-        Transfer the input track's mixing style to that of reference's by uploading files or downloading via YouTube links. This demo page also provides mastering style transfer, which can be interpreted as an extended work of [this paper](https://huggingface.co/papers/2202.08520).
+            - [project page](https://jhtonykoo.github.io/MixingStyleTransfer/)
+            - [GitHub](https://github.com/jhtonyKoo/music_mixing_style_transfer)
+            - [supplementary](https://pale-cicada-946.notion.site/Music-Mixing-Style-Transfer-A-Contrastive-Learning-Approach-to-Disentangle-Audio-Effects-Supplemen-e6eccd9a431a4a8fa4fdd5adb2d3f219)
         """
     )
     with gr.Group():
@@ -169,23 +161,6 @@ with gr.Blocks() as demo:
                 outputs=[output_mix],
             )
 
-    
-    with gr.Group():
-        gr.HTML(
-            """
-            <div> <h3> <center> Mastering Style Transfer. Perform mastering style transfer using the FXencoder.</h3> </div>
-            """
-        )
-        with gr.Column():
-            inference_mastering_btn = gr.Button("Run Mastering Style Transfer")
-        with gr.Row():
-            output_master = gr.Audio(label="mastering style transferred music track")
-            inference_mastering_btn.click(
-                inference_mastering,
-                inputs=[file_uploaded_in, file_uploaded_ref],
-                outputs=[output_master],
-            )
-            
 
     
 if __name__ == "__main__":