Upload app.py

app.py

@@ -1,5 +1,5 @@
-import spaces
-import zero
+# import spaces
+# import zero
 import gradio as gr
 import sys
 import threading
@@ -15,6 +15,33 @@ from inference import inference_patch
 from convert import abc2xml, xml2, pdf2img
 
 
+title_html = """
+<div class="title-container">
+    <h1 class="title-text">NotaGen</h1> &nbsp;
+        <!-- ArXiv -->
+        <a href="https://arxiv.org/abs/2502.18008">
+            <img src="https://img.shields.io/badge/NotaGen_Paper-ArXiv-%23B31B1B?logo=arxiv&logoColor=white" alt="Paper">
+        </a>
+        &nbsp;
+        <!-- GitHub -->
+        <a href="https://github.com/ElectricAlexis/NotaGen">
+            <img src="https://img.shields.io/badge/NotaGen_Code-GitHub-%23181717?logo=github&logoColor=white" alt="GitHub">
+        </a>
+        &nbsp;
+        <!-- HuggingFace -->
+        <a href="https://huggingface.co/ElectricAlexis/NotaGen">
+            <img src="https://img.shields.io/badge/NotaGen_Weights-HuggingFace-%23FFD21F?logo=huggingface&logoColor=white" alt="Weights">
+        </a>
+        &nbsp;
+        <!-- Web Demo -->
+        <a href="https://electricalexis.github.io/notagen-demo/">
+            <img src="https://img.shields.io/badge/NotaGen_Demo-Web-%23007ACC?logo=google-chrome&logoColor=white" alt="Demo">
+        </a>
+</div>
+<bp>
+<p style="font-size: 1.2em;">NotaGen is a model for generating sheet music in ABC notation format. Select a period, composer, and instrumentation to generate classical-style music!</p>
+"""
+
 # Read prompt combinations
 with open('prompts.txt', 'r') as f:
     prompts = f.readlines()
@@ -143,7 +170,7 @@ def update_page(direction, data):
     return new_image, prev_btn_state, next_btn_state, data
 
 
-@spaces.GPU(duration=600)
+# @spaces.GPU(duration=600)
 def generate_music(period, composer, instrumentation):
     """
     Must ensure each yield returns the same number of values.
@@ -324,10 +351,50 @@ button[size="sm"] {
     box-shadow: 0 2px 8px rgba(0,0,0,0.1);
 }
 
+/* Social icons styling */
+.title-container {
+    display: flex;
+    align-items: center;
+    gap: 15px;
+    margin-bottom: 10px;
+}
+
+.title-text {
+    margin: 0;
+    font-size: 1.8em;
+}
+
+.social-icons {
+    display: flex;
+    gap: 10px;
+}
+
+.social-icon {
+    display: inline-flex;
+    align-items: center;
+    justify-content: center;
+    width: 32px;
+    height: 32px;
+    border-radius: 50%;
+    background-color: #f5f5f5;
+    text-decoration: none;
+    transition: transform 0.2s, background-color 0.2s;
+}
+
+.social-icon:hover {
+    transform: scale(1.1);
+    background-color: #e0e0e0;
+}
+
+.social-icon img {
+    width: 20px;
+    height: 20px;
+}
+
 """
 
 with gr.Blocks(css=css) as demo:
-    gr.Markdown("## NotaGen")
+    gr.HTML(title_html)
 
     # For storing PDF page count, current page and other information
     pdf_state = gr.State()