Update app.py

app.py

@@ -243,10 +243,12 @@ def on_num_iterations_change(n, prompt_2):
     return gr.update(value=f"Total length: {total_length:.2f} seconds")
 
 with gr.Blocks() as app:
-    gr.Markdown("## Riffusion")
+    gr.Markdown("## Riffusion Demo")
     gr.Markdown("""Generate audio using the [Riffusion](https://huggingface.co/riffusion/riffusion-model-v1) model.<br>
                 In single prompt mode you can generate up to ~1 minute of audio with smooth transitions between sections. (beta)<br>
                 Bi-prompt mode interpolates between two prompts. It can generate up to ~2 minutes of audio, but the transitions between sections are more abrupt.""")
+    gr.Markdown(f"""Running on {"**GPU 🔥**" if torch.cuda.is_available() else f"**CPU 🥶**. For faster inference it is recommended to **upgrade to GPU in space's Settings**"}<br>
+                [![Duplicate Space](https://bit.ly/3gLdBN6)](https://huggingface.co/spaces/$space_id?duplicate=true)""")
     
     with gr.Row():
         with gr.Group():
@@ -261,7 +263,7 @@ with gr.Blocks() as app:
                 seed = gr.Slider(minimum=0, maximum=4294967295, value=0, step=1, label="Seed (0 for random)")
 
             info = gr.Markdown()
-            btn_generate = gr.Button(value="Generate")
+            btn_generate = gr.Button(value="Generate").style(full_width=True)
         with gr.Column():
             video = gr.Video()
 
@@ -288,5 +290,16 @@ with gr.Blocks() as app:
             ["soulful R&B love song", "", "marim", 10],
             ["country western twangy guitar", "", "agile", 10]],
         inputs=[prompt_1, prompt_2, feel, num_iterations])
+    
+    gr.HTML("""
+    <div style="border-top: 1px solid #303030;">
+      <br>
+      <p>Space by:<br>
+      <a href="https://twitter.com/hahahahohohe"><img src="https://img.shields.io/twitter/follow/hahahahohohe?label=%40anzorq&style=social" alt="Twitter Follow"></a><br>
+      <a href="https://github.com/qunash"><img alt="GitHub followers" src="https://img.shields.io/github/followers/qunash?style=social" alt="Github Follow"></a></p><br>
+      <a href="https://www.buymeacoffee.com/anzorq" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/v2/default-yellow.png" alt="Buy Me A Coffee" style="height: 24px !important;width: 81px !important;" ></a><br><br>
+      <p><img src="https://visitor-badge.glitch.me/badge?page_id=anzorq.riffusion-demo" alt="visitors"></p>
+    </div>
+    """)
 
 app.launch()