Update app.py
Browse files
app.py
CHANGED
|
@@ -5,6 +5,11 @@ import torch
|
|
| 5 |
import torchaudio
|
| 6 |
from io import BytesIO
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
@st.cache_resource
|
| 9 |
def load_model():
|
| 10 |
model = MusicGen.get_pretrained("facebook/musicgen-small")
|
|
@@ -41,6 +46,8 @@ def save_audio_to_bytes(samples: torch.Tensor):
|
|
| 41 |
audio_buffer.seek(0) # Move to the start of the buffer
|
| 42 |
return audio_buffer
|
| 43 |
|
|
|
|
|
|
|
| 44 |
video_background = """
|
| 45 |
<style>
|
| 46 |
.video-container {
|
|
@@ -96,10 +103,7 @@ def main():
|
|
| 96 |
)
|
| 97 |
st.write("Generating your music... please wait.")
|
| 98 |
|
| 99 |
-
|
| 100 |
-
page_icon=":musical_note:",
|
| 101 |
-
page_title="Music Gen"
|
| 102 |
-
)
|
| 103 |
|
| 104 |
def main():
|
| 105 |
st.title("Your Music")
|
|
|
|
| 5 |
import torchaudio
|
| 6 |
from io import BytesIO
|
| 7 |
|
| 8 |
+
st.set_page_config(
|
| 9 |
+
page_icon=":musical_note:",
|
| 10 |
+
page_title="Music Gen"
|
| 11 |
+
)
|
| 12 |
+
|
| 13 |
@st.cache_resource
|
| 14 |
def load_model():
|
| 15 |
model = MusicGen.get_pretrained("facebook/musicgen-small")
|
|
|
|
| 46 |
audio_buffer.seek(0) # Move to the start of the buffer
|
| 47 |
return audio_buffer
|
| 48 |
|
| 49 |
+
|
| 50 |
+
|
| 51 |
video_background = """
|
| 52 |
<style>
|
| 53 |
.video-container {
|
|
|
|
| 103 |
)
|
| 104 |
st.write("Generating your music... please wait.")
|
| 105 |
|
| 106 |
+
|
|
|
|
|
|
|
|
|
|
| 107 |
|
| 108 |
def main():
|
| 109 |
st.title("Your Music")
|