Update app.py

app.py

@@ -31,8 +31,8 @@ def load_model(version):
     return MusicGen.get_pretrained(version)
 
 
-def predict(uploaded_image, melody, duration):
-    text = "à remplacer"
+def predict(music_prompt, melody, duration):
+    text = music_prompt
     global MODEL
     topk = int(250)
     if MODEL is None or MODEL.name != "melody":
@@ -94,7 +94,7 @@ with gr.Blocks() as demo:
                 #split_drums = gr.Audio(type="filepath", label="Drums")
                 #split_others = gr.Audio(type="filepath", label="Other")
             with gr.Row():
-                uploaded_image = gr.Image(label="Input Image", interactive=True, source="upload", type="filepath")
+                music_prompt = gr.Textbox(label="Musical Prompt", interactive=True)
                 melody = gr.Audio(source="upload", type="numpy", label="Melody Condition (optional)", interactive=True)
             with gr.Row():
                 submit = gr.Button("Submit")
@@ -110,7 +110,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             output = gr.Audio(label="Generated Music")
     load_sound_btn.click(split_process, inputs=[uploaded_sound], outputs=[melody])
-    submit.click(predict, inputs=[uploaded_image, melody, duration], outputs=[output])
+    submit.click(predict, inputs=[music_prompt, melody, duration], outputs=[output])
     
     gr.Markdown(
         """