Update app.py

app.py

@@ -343,98 +343,73 @@ div#component-11 > .gallery > .gallery-item > .container > img {
 }
 """
 
-
-
 with gr.Blocks(css=css) as demo:
     
     with gr.Column(elem_id="col-container"):
     
         gr.HTML(f"""
-        <h2 style="text-align: center;">{demo_title}</h2>
-        <p style="text-align: center;">{description}</p>
+        <h2 style="text-align: center;">Music Generation Demo</h2>
+        <p style="text-align: center;">Extracts an image from a video to generate a description and creates music based on it.</p>
         """)
         
         with gr.Row():
             
             with gr.Column():
-                video_in = gr.Video(sources=["upload"], label="Video input")
+                video_in = gr.Video(sources=["upload"], label="Video Input")
                 
                 with gr.Row():
                     
                     chosen_model = gr.Dropdown(
-                        label = "Choose a model",
-                        choices = [
+                        label="Choose a Model",
+                        choices=[
                             "MAGNet",
                             "AudioLDM-2",
                             "Riffusion",
                             "Mustango",
                             "MusicGen"
                         ],
-                        value = None,
-                        filterable = False
+                        value=None,
+                        filterable=False
                     )
                     
                     check_status = gr.Textbox(
-                        label="API status",
+                        label="API Status",
                         interactive=False
                     )
                 
-                submit_btn = gr.Button("Make music from my shot !")
-
-                gr.Examples(
-                    examples = [
-                        "vid_examples/big-sur.mp4",
-                        "vid_examples/snow-dogs.mp4",
-                        "vid_examples/ocean-turtle.mp4"
-                    ],
-                    inputs = [video_in],
-                    cache_examples=False   
-                )
+                submit_btn = gr.Button("Generate Music")
             
             with gr.Column():
             
                 caption = gr.Textbox(
-                    label = "Inspirational musical prompt",
-                    interactive = False,
-                    elem_id = "inspi-prompt"
+                    label="Inspirational Musical Prompt",
+                    interactive=False,
+                    elem_id="inspi-prompt"
                 )
                 
-                retry_btn = gr.Button("Retry with edited prompt", visible=False)
+                retry_btn = gr.Button("Retry with Edited Prompt", visible=False)
                 
-                result = gr.Audio(
-                    label = "Music"
-                )
-
-                video_o = gr.Video(label="Video with soundFX")
+                result = gr.Audio(label="Music")
+                video_o = gr.Video(label="Video with SoundFX")
         
-
     chosen_model.change(
-        fn = check_api,
-        inputs = chosen_model,
-        outputs = check_status,
-        queue = False
+        fn=check_api,
+        inputs=chosen_model,
+        outputs=check_status,
+        queue=False
     )
 
     retry_btn.click(
-        fn = retry,
-        inputs = [video_in, chosen_model, caption],
-        outputs = [result, video_o]
+        fn=retry,
+        inputs=[video_in, chosen_model, caption],
+        outputs=[result, video_o]
     )
     
     submit_btn.click(
-        fn = infer,
-        inputs = [
-            video_in,
-            chosen_model,
-            check_status
-        ],
-        outputs =[
-            caption,
-            retry_btn,
-            result,
-            video_o
-        ],
-        concurrency_limit = 4
+        fn=infer,
+        inputs=[video_in, chosen_model, check_status],
+        outputs=[caption, retry_btn, result, video_o],
+        concurrency_limit=4
     )
 
 demo.queue(max_size=16).launch(show_api=False)