chong.zhang commited on
Commit
f107ce2
·
1 Parent(s): 4ab7f58
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
4
 
5
  def get_args(
6
- task, text="", audio=None, model_name="InspireMusic-1.5B-Long",
7
  chorus="intro", fast=False, fade_out=True, trim=False,
8
  output_sample_rate=48000, max_generate_audio_seconds=30.0):
9
  # This function constructs the arguments required for InspireMusic
@@ -69,7 +69,7 @@ with gr.Blocks() as demo:
69
  with gr.Row():
70
  task = gr.Radio(["text-to-music", "continuation"], label="Select Task",
71
  value="text-to-music")
72
- model_name = gr.Dropdown(["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-1.5B-24kHz", "InspireMusic-Base", "InspireMusic-Base-24kHz"], label="Model Name", value="InspireMusic-1.5B-Long")
73
 
74
  text_input = gr.Textbox(label="Input Text (For Text-to-Music Task)")
75
  audio_input = gr.Audio(label="Input Audio (For Music Continuation Task)",
 
3
  from inspiremusic.cli.inference import InspireMusicUnified, set_env_variables
4
 
5
  def get_args(
6
+ task, text="", audio=None, model_name="InspireMusic-Base",
7
  chorus="intro", fast=False, fade_out=True, trim=False,
8
  output_sample_rate=48000, max_generate_audio_seconds=30.0):
9
  # This function constructs the arguments required for InspireMusic
 
69
  with gr.Row():
70
  task = gr.Radio(["text-to-music", "continuation"], label="Select Task",
71
  value="text-to-music")
72
+ model_name = gr.Dropdown(["InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-1.5B-24kHz", "InspireMusic-Base", "InspireMusic-Base-24kHz"], label="Model Name", value="InspireMusic-Base")
73
 
74
  text_input = gr.Textbox(label="Input Text (For Text-to-Music Task)")
75
  audio_input = gr.Audio(label="Input Audio (For Music Continuation Task)",