Upload 2 files

app.py

@@ -0,0 +1,37 @@
+from audiocraft.models import MusicGen
+import streamlit as st
+import os 
+import torch
+import torchaudio
+import numpy as np
+import base64
+
+@st.cache_resource
+def load_model():
+    model=MusicGen.get_pretrained("facebook/musicgen-small")
+    return model
+
+st.set_page_config(
+    page_icon=":musical_note:",
+    page_title="Music Gen"
+)
+
+def main():
+    st.title("Your Music")
+
+    with st.expander("See Explanation"):
+        st.write("App is developed by using Meta's Audiocraft Music Gen model. Write your text and we will generate audio")
+    text_area=st.text_area("Enter description")
+    time_slider=st.slider("Select time duration(s)",2,5,20)
+    
+    if text_area and time_slider:
+        st.json(
+            {
+                "Description":text_area,
+                "Selected duration:":time_slider
+            }
+        )
+        st.subheader("Generated Music")
+
+if __name__=="__main__":
+    main()

requirements.txt

@@ -0,0 +1,28 @@
+# please make sure you have already a pytorch install that is cuda enabled!
+av==11.0.0
+einops
+flashy>=0.0.1
+hydra-core>=1.1
+hydra_colorlog
+julius
+num2words
+numpy<2.0.0
+sentencepiece
+spacy==3.7.6
+torch==2.1.0
+torchaudio>=2.0.0,<2.1.2
+huggingface_hub
+tqdm
+transformers>=4.31.0  # need Encodec there.
+xformers<0.0.23
+demucs
+librosa
+soundfile
+gradio
+torchmetrics
+encodec
+protobuf
+torchvision==0.16.0
+torchtext==0.16.0
+pesq
+pystoi