Update app.py

app.py

@@ -21,7 +21,6 @@ model = Pop2PianoForConditionalGeneration.from_pretrained("sweetcocoa/pop2piano"
 processor = Pop2PianoProcessor.from_pretrained("sweetcocoa/pop2piano")
 composers = model.generation_config.composer_to_feature_token.keys()
 
-
 def get_audio_from_yt_video(yt_link):
     try:
         yt = pt.YouTube(yt_link)
@@ -59,7 +58,7 @@ def prepare_output_file(tokenizer_output, sr):
     
     return wav_output, wav_output, midi_output
 
-def get_stereo(pop_path, midi, pop_scale=0.99):
+def get_stereo(pop_path, midi, pop_scale=0.5):
     pop_y, sr = librosa.load(pop_path, sr=None)
     midi_y, _ = librosa.load(midi.name, sr=None)
 
@@ -134,13 +133,13 @@ with block:
             <div> <h3> <center> Get the Stereo Mix from the Pop Music and Generated MIDI </h3> </div>
             """
         )
-        gr.Slider(0, 1, value=0.5, label="Choose the ratio between Pop and MIDI", info="1.0 = Only Pop, 0.0=Only MIDI"),
+        pop_scale = gr.Slider(0, 1, value=0.5, label="Choose the ratio between Pop and MIDI", info="1.0 = Only Pop, 0.0=Only MIDI", interactive=True),
         stereo_btn = gr.Button("Get Stereo Mix")
         with gr.Row():
             stereo_mix1 = gr.Audio(label="Listen to the Stereo Mix")
             stereo_mix2 = gr.File(label="Download the Stereo Mix")
         
-        stereo_btn.click(get_stereo, inputs=[file_uploaded, wav_output2], outputs=[stereo_mix1, stereo_mix2])
+        stereo_btn.click(get_stereo, inputs=[file_uploaded, wav_output2, pop_scale[0]], outputs=[stereo_mix1, stereo_mix2])
                 
     with gr.Group():
         gr.Examples([
@@ -167,8 +166,6 @@ with block:
                     <center><p><a href="http://sweetcocoa.github.io/pop2piano_samples" style="text-decoration: underline;" target="_blank">Project Page</a> 
                     <center><a href="https://huggingface.co/docs/transformers/main/model_doc/pop2piano" style="text-decoration: underline;" target="_blank">HuggingFace Model Docs</a>
                     <center><a href="https://github.com/sweetcocoa/pop2piano" style="text-decoration: underline;" target="_blank">Github</a>
-                    
-                    
                     </p>
         </div>
         """