update

README.md

@@ -1,6 +1,6 @@
 ---
-title: FluxMusic
-emoji: 💻
+title: OpenMusic
+emoji: 🎶
 colorFrom: indigo
 colorTo: yellow
 sdk: gradio

app.py

@@ -6,11 +6,11 @@ import sys
 
 # we will clone the repo and install the dependencies
 # NOTE: Still fixing bugs, not release, do not try :) !
-os.system('pip install -r qa_mdt/requirements.txt')
-os.system('pip install xformers==0.0.26.post1')
-os.system('pip install torchlibrosa==0.0.9 librosa==0.9.2')
-os.system('pip install -q pytorch_lightning==2.1.3 torchlibrosa==0.0.9 librosa==0.9.2 ftfy==6.1.1 braceexpand')
-os.system('pip install torch==2.3.0+cu121 torchvision==0.18.0+cu121 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121')
+# os.system('pip install -r qa_mdt/requirements.txt')
+# os.system('pip install xformers==0.0.26.post1')
+# os.system('pip install torchlibrosa==0.0.9 librosa==0.9.2')
+# os.system('pip install -q pytorch_lightning==2.1.3 torchlibrosa==0.0.9 librosa==0.9.2 ftfy==6.1.1 braceexpand')
+# os.system('pip install torch==2.3.0+cu121 torchvision==0.18.0+cu121 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu121')
 
 # only then import the necessary modules from qa_mdt
 from qa_mdt.pipeline import MOSDiffusionPipeline
@@ -21,7 +21,8 @@ pipe = MOSDiffusionPipeline()
 # this runs the pipeline with user input and saves the output as 'awesome.wav'
 @spaces.GPU(duration=120)
 def generate_waveform(description):
-    pipe(description)
+    high_quality_description = "high quality " + description
+    pipe(high_quality_description)
 
     generated_file_path = "./awesome.wav"
 
@@ -34,19 +35,15 @@ def generate_waveform(description):
 intro = """
 # 🎶 OpenMusic: AI-Powered Music Diffusion 🎶
 
-![OpenMusic Banner](./banner.png)
-
 Welcome to **OpenMusic**, a next-gen diffusion model designed to generate high-quality audio from text descriptions! 
 
 Simply enter a description of the music you'd like to hear, and our AI will generate it for you.
 
----
-
 ### Powered by:
 
-- [GitHub Repository](https://github.com/ivcylc/qa-mdt) by [@changli](https://github.com/ivcylc) 🎓.
-- Introduced in this [Paper](https://arxiv.org/pdf/2405.15863)
-- Hugging Face Diffusers Implementation 🧨 (Super easy to use): [Model](https://huggingface.co/jadechoghari/qa_mdt) by [@jadechoghari](https://github.com/jadechoghari) 🤗.
+- [GitHub](https://github.com/ivcylc/qa-mdt) [@changli](https://github.com/ivcylc) 🎓.
+- [Paper](https://arxiv.org/pdf/2405.15863)
+- [[HuggingFace](https://huggingface.co/jadechoghari/qa_mdt) [@jadechoghari](https://github.com/jadechoghari) 🤗.
 
 ---