Update app UI

app.py

@@ -29,7 +29,7 @@ if os.path.isfile("transfer.pth") == False:
         "orchestra.wav",
     )
 
-device = "cuda:0" if torch.cuda.is_available() else "cpu"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 example_list = ["folk.wav", "electronic.mp3", "orchestra.wav"]
 model = BartCaptionModel(max_length=128)
@@ -96,7 +96,13 @@ def captioning(audio_path):
     return inference
 
 
-title = "Interactive demo: Music Captioning 🤖🎵"
+def load_css():
+    with open("static/css/musicapp.css", "r") as file:
+        css_content = file.read()
+    return css_content
+
+
+title = "Capabara - Interactive demo: Music Captioning 🤖🎵"
 description = """
 <p style='text-align: center'> LP-MusicCaps: LLM-Based Pseudo Music Captioning</p> 
 <p style='text-align: center'> SeungHeon Doh, Keunwoo Choi, Jongpil Lee, Juhan Nam, ISMIR 2023</p> 
@@ -115,9 +121,11 @@ demo = gr.Interface(
     ],
     examples=example_list,
     title=title,
+    theme=gr.themes.Default(font=[gr.themes.GoogleFont("Work Sans"), "sans-serif"]),
     description=description,
     article=article,
     cache_examples=False,
+    css=load_css(),
 )
 
 demo.launch()

requirements.txt

@@ -1,7 +1,79 @@
---extra-index-url https://download.pytorch.org/whl/cpu
-torch==1.13.1
-torchaudio==0.13.1
-transformers==4.42.0
-librosa>=0.8
+aiofiles==23.2.1
+annotated-types==0.7.0
+anyio==4.8.0
+audioread==3.0.1
+certifi==2024.12.14
+cffi==1.17.1
+charset-normalizer==3.4.1
+click==8.1.8
+colorama==0.4.6
+decorator==5.1.1
+fastapi==0.115.6
+ffmpy==0.5.0
+filelock==3.16.1
+fsspec==2024.12.0
 gradio==5.8.0
-numpy<2
+gradio_client==1.5.1
+h11==0.14.0
+httpcore==1.0.7
+httpx==0.28.1
+huggingface-hub==0.27.1
+idna==3.10
+Jinja2==3.1.5
+joblib==1.4.2
+lazy_loader==0.4
+librosa==0.10.2.post1
+llvmlite==0.43.0
+markdown-it-py==3.0.0
+MarkupSafe==2.1.5
+mdurl==0.1.2
+mpmath==1.3.0
+msgpack==1.1.0
+networkx==3.4.2
+numba==0.60.0
+numpy==1.26.4
+orjson==3.10.14
+packaging==24.2
+pandas==2.2.3
+pillow==11.1.0
+platformdirs==4.3.6
+pooch==1.8.2
+pycparser==2.22
+pydantic==2.10.5
+pydantic_core==2.27.2
+pydub==0.25.1
+Pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-multipart==0.0.20
+pytz==2024.2
+PyYAML==6.0.2
+regex==2024.11.6
+requests==2.32.3
+rich==13.9.4
+ruff==0.9.1
+safehttpx==0.1.6
+safetensors==0.5.2
+scikit-learn==1.6.1
+scipy==1.15.1
+semantic-version==2.10.0
+setuptools==75.8.0
+shellingham==1.5.4
+six==1.17.0
+sniffio==1.3.1
+soundfile==0.13.0
+soxr==0.5.0.post1
+starlette==0.41.3
+sympy==1.13.1
+threadpoolctl==3.5.0
+tokenizers==0.19.1
+tomlkit==0.13.2
+torch==2.5.1
+torchaudio==2.5.1
+tqdm==4.67.1
+transformers==4.42.0
+typer==0.15.1
+typing_extensions==4.12.2
+tzdata==2024.2
+urllib3==2.3.0
+uvicorn==0.34.0
+websockets==14.1

static/css/musicapp.css

@@ -0,0 +1,63 @@
+body {
+  background: #0c4a6e !important;
+}
+
+body > gradio-app {
+  background: transparent !important;
+}
+
+h1,
+h4,
+li,
+p {
+  color: whitesmoke;
+}
+
+h1 {
+  text-align: center;
+}
+
+a {
+  color: #f2762e;
+}
+
+#capa-logo {
+  width: 11rem;
+  display: block;
+  margin: 0 auto 0 auto;
+}
+
+#capa-logo > div.image-container > button {
+  cursor: default;
+}
+
+#capa-logo > div.image-container > button:hover {
+  background-color: transparent !important;
+}
+
+#capa-logo > div.image-container > div {
+  display: none;
+}
+
+#capa-title > div.html-container {
+  padding: 0;
+}
+
+.secondary {
+  background: #f2762e;
+  border-color: #f2762e;
+  color: white;
+}
+
+.secondary:hover {
+  background: #fc9f53;
+  border-color: #fc9f53;
+}
+
+.label {
+  color: whitesmoke !important;
+}
+
+.gallery > button > div {
+  background: whitesmoke !important;
+}