Update demos/melodyflow_app.py

demos/melodyflow_app.py

@@ -29,7 +29,7 @@ SPACE_ID = os.environ.get('SPACE_ID', '')
 MODEL_PREFIX = os.environ.get('MODEL_PREFIX', 'facebook/')
 IS_HF_SPACE = (MODEL_PREFIX + "MelodyFlow") in SPACE_ID
 MAX_BATCH_SIZE = 12
-N_REPEATS = 3
+N_REPEATS = 1
 INTERRUPTING = False
 MBD = None
 # We have to wrap subprocess call to clean a bit the log when using gr.make_waveform
@@ -214,7 +214,7 @@ def toggle_audio_src(choice):
 
 def toggle_melody(melody):
     if melody is None:
-        return gr.update(value=MIDPOINT)
+        return gr.update(value=EULER)
     else:
         return gr.update(value=EULER)
 
@@ -222,14 +222,14 @@ def toggle_melody(melody):
 def toggle_solver(solver, melody):
     if melody is None:
         if solver == MIDPOINT:
-            return gr.update(value=64.0, minimum=2, maximum=128.0, step=2.0), gr.update(interactive=False, value=1.0), gr.update(interactive=False, value=False), gr.update(interactive=False, value=0.0), gr.update(interactive=True, value=30.0)
+            return gr.update(value=50.0, minimum=2, maximum=128.0, step=2.0), gr.update(interactive=False, value=1.0), gr.update(interactive=False, value=False), gr.update(interactive=False, value=0.0), gr.update(interactive=True, value=10.0)
         else:
-            return gr.update(value=64.0, minimum=1, maximum=128.0, step=1.0), gr.update(interactive=False, value=1.0), gr.update(interactive=False, value=False), gr.update(interactive=False, value=0.0), gr.update(interactive=True, value=30.0)
+            return gr.update(value=50.0, minimum=1, maximum=128.0, step=1.0), gr.update(interactive=False, value=1.0), gr.update(interactive=False, value=False), gr.update(interactive=False, value=0.0), gr.update(interactive=True, value=10.0)
     else:
         if solver == MIDPOINT:
-            return gr.update(value=128, minimum=4.0, maximum=256.0, step=4.0), gr.update(interactive=True, value=0.0), gr.update(interactive=False, value=False), gr.update(interactive=False, value=0.0), gr.update(interactive=False, value=0.0)
+            return gr.update(value=100, minimum=4.0, maximum=256.0, step=4.0), gr.update(interactive=True, value=0.0), gr.update(interactive=False, value=False), gr.update(interactive=False, value=0.0), gr.update(interactive=False, value=0.0)
         else:
-            return gr.update(value=125, minimum=5.0, maximum=250.0, step=5.0), gr.update(interactive=True, value=0.0), gr.update(interactive=True, value=True), gr.update(interactive=True, value=0.2), gr.update(interactive=False, value=0.0)
+            return gr.update(value=50, minimum=5.0, maximum=250.0, step=5.0), gr.update(interactive=True, value=0.0), gr.update(interactive=True, value=True), gr.update(interactive=True, value=0.0), gr.update(interactive=False, value=0.0)
 
 def ui_local(launch_kwargs):
     with gr.Blocks() as interface:
@@ -257,16 +257,16 @@ def ui_local(launch_kwargs):
                     model_path = gr.Text(label="Model Path (custom models)")
                 with gr.Row():
                     solver = gr.Radio([EULER, MIDPOINT],
-                                      label="ODE Solver", value=MIDPOINT, interactive=True)
+                                      label="ODE Solver", value=EULER, interactive=True)
                     steps = gr.Slider(label="Inference steps", minimum=2.0, maximum=128.0,
-                                      step=2.0, value=128.0, interactive=True)
-                    duration = gr.Slider(label="Duration", minimum=1.0, maximum=30.0, value=30.0, interactive=True)
+                                      step=2.0, value=50.0, interactive=True)
+                    duration = gr.Slider(label="Duration", minimum=1.0, maximum=30.0, value=10.0, interactive=True)
                 with gr.Row():
                     target_flowstep = gr.Slider(label="Target Flow step", minimum=0.0,
                                                 maximum=1.0, value=0.0, interactive=False)
                     regularize = gr.Checkbox(label="Regularize", value=False, interactive=False)
                     regularization_strength = gr.Slider(
-                        label="Regularization Strength", minimum=0.0, maximum=1.0, value=0.2, interactive=False)
+                        label="Regularization Strength", minimum=0.0, maximum=1.0, value=0.0, interactive=False)
             with gr.Column():
                 audio_outputs = [
                     gr.Audio(label=f"Generated Audio - variation {i+1}", type='filepath', show_download_button=False, show_share_button=False) for i in range(N_REPEATS)]
@@ -290,22 +290,22 @@ def ui_local(launch_kwargs):
                 [
                     (MODEL_PREFIX + "melodyflow-t24-30secs"),
                     "80s electronic track with melodic synthesizers, catchy beat and groovy bass.",
-                    MIDPOINT,
-                    64,
+                    EULER,
+                    50,
                     1.0,
                     False,
                     0.0,
-                    30.0,
+                    10.0,
                     None,
                 ],
                 [
                     (MODEL_PREFIX + "melodyflow-t24-30secs"),
                     "A cheerful country song with acoustic guitars accompanied by a nice piano melody.",
                     EULER,
-                    125,
+                    50,
                     0.0,
                     True,
-                    0.2,
+                    0.0,
                     -1.0,
                     "./assets/bolero_ravel.mp3",
                 ],
@@ -376,16 +376,16 @@ def ui_hf(launch_kwargs):
                                      label="Model", value=(MODEL_PREFIX + "melodyflow-t24-30secs"), interactive=True)
                 with gr.Row():
                     solver = gr.Radio([EULER, MIDPOINT],
-                                      label="ODE Solver", value=MIDPOINT, interactive=True)
+                                      label="ODE Solver", value=EULER, interactive=True)
                     steps = gr.Slider(label="Inference steps", minimum=2.0, maximum=128.0,
-                                      step=2.0, value=128.0, interactive=True)
-                    duration = gr.Slider(label="Duration", minimum=1.0, maximum=30.0, value=30.0, interactive=True)
+                                      step=2.0, value=50.0, interactive=True)
+                    duration = gr.Slider(label="Duration", minimum=1.0, maximum=30.0, value=10.0, interactive=True)
                 with gr.Row():
                     target_flowstep = gr.Slider(label="Target Flow step", minimum=0.0,
                                                 maximum=1.0, value=0.0, interactive=False)
                     regularize = gr.Checkbox(label="Regularize", value=False, interactive=False)
                     regularization_strength = gr.Slider(
-                        label="Regularization Strength", minimum=0.0, maximum=1.0, value=0.2, interactive=False)
+                        label="Regularization Strength", minimum=0.0, maximum=1.0, value=0.0, interactive=False)
             with gr.Column():
                 audio_outputs = [
                     gr.Audio(label=f"Generated Audio - variation {i+1}", type='filepath', show_download_button=False, show_share_button=False) for i in range(N_REPEATS)]
@@ -408,22 +408,22 @@ def ui_hf(launch_kwargs):
                 [
                     (MODEL_PREFIX + "melodyflow-t24-30secs"),
                     "80s electronic track with melodic synthesizers, catchy beat and groovy bass.",
-                    MIDPOINT,
-                    64,
+                    EULER,
+                    50,
                     1.0,
                     False,
                     0.0,
-                    30.0,
+                    10.0,
                     None,
                 ],
                 [
                     (MODEL_PREFIX + "melodyflow-t24-30secs"),
                     "A cheerful country song with acoustic guitars accompanied by a nice piano melody.",
                     EULER,
-                    125,
+                    50,
                     0.0,
                     True,
-                    0.2,
+                    0.0,
                     -1.0,
                     "./assets/bolero_ravel.mp3",
                 ],
@@ -509,4 +509,4 @@ if __name__ == "__main__":
     if IS_HF_SPACE:
         ui_hf(launch_kwargs)
     else:
-        ui_local(launch_kwargs)
+        ui_local(launch_kwargs)