Update demos/musicgen_app.py

demos/musicgen_app.py

@@ -241,7 +241,7 @@ def toggle_diffusion(choice):
         return [gr.update(visible=False)]
 
 
-def ui_full(launch_kwargs):
+def ui_full():
     with gr.Blocks() as interface:
         gr.Markdown(
             """
@@ -289,12 +289,13 @@ def ui_full(launch_kwargs):
                     cfg_coef = gr.Number(label="Classifier Free Guidance", value=3.0, interactive=True)
             with gr.Column():
                 audio_output = gr.Textbox(label="Generated Music (wav)")
-                audio_diffusion = gr.Textbox(label="MultiBand Diffusion Decoder (wav)")
-        submit.click(toggle_diffusion, decoder, [audio_diffusion], queue=False,
-                     show_progress=False).then(predict_full, inputs=[secret_token, model, model_path, decoder, text, melody, duration, topk, topp,
+        
+        submit.click(
+            fn=predict_full,
+             inputs=[secret_token, model, model_path, decoder, text, melody, duration, topk, topp,
                                                                      temperature, cfg_coef],
-                                               outputs=[audio_output])
-        radio.change(toggle_audio_src, radio, [melody], queue=False, show_progress=False)
+            outputs=audio_output,
+            api_name="run")
 
         gr.Markdown(
             """
@@ -341,52 +342,11 @@ def ui_full(launch_kwargs):
             """
         )
 
-        interface.queue().launch(**launch_kwargs)
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser()
-    parser.add_argument(
-        '--listen',
-        type=str,
-        default='0.0.0.0' if 'SPACE_ID' in os.environ else '127.0.0.1',
-        help='IP to listen on for connections to Gradio',
-    )
-    parser.add_argument(
-        '--username', type=str, default='', help='Username for authentication'
-    )
-    parser.add_argument(
-        '--password', type=str, default='', help='Password for authentication'
-    )
-    parser.add_argument(
-        '--server_port',
-        type=int,
-        default=0,
-        help='Port to run the server listener on',
-    )
-    parser.add_argument(
-        '--inbrowser', action='store_true', help='Open in browser'
-    )
-    parser.add_argument(
-        '--share', action='store_true', help='Share the gradio UI'
-    )
-
-    args = parser.parse_args()
-
-    launch_kwargs = {}
-    launch_kwargs['server_name'] = args.listen
-
-    if args.username and args.password:
-        launch_kwargs['auth'] = (args.username, args.password)
-    if args.server_port:
-        launch_kwargs['server_port'] = args.server_port
-    if args.inbrowser:
-        launch_kwargs['inbrowser'] = args.inbrowser
-    if args.share:
-        launch_kwargs['share'] = args.share
-
-    logging.basicConfig(level=logging.INFO, stream=sys.stderr)
-
-    # Show the interface
-    # we preload the model to avoid a timeout on the first request
-    load_model('facebook/musicgen-stereo-large')
-    ui_full(launch_kwargs)
+        interface.queue(max_size=12).launch()
+
+logging.basicConfig(level=logging.INFO, stream=sys.stderr)
+
+# Show the interface
+# we preload the model to avoid a timeout on the first request
+load_model('facebook/musicgen-stereo-large')
+ui_full()