Duplicate from riffusion/riffusion-playground

Co-authored-by: Hayk Martiros <[email protected]>

.gitattributes

@@ -0,0 +1,34 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "riffusion"]
+	path = riffusion
+	url = https://github.com/riffusion/riffusion

README.md

@@ -0,0 +1,14 @@
+---
+title: Riffusion Playground
+emoji: 📚
+colorFrom: red
+colorTo: purple
+sdk: streamlit
+sdk_version: 1.15.2
+app_file: app.py
+pinned: false
+license: mit
+duplicated_from: riffusion/riffusion-playground
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,36 @@
+"""
+Shim layer for using the riffusion playground streamlit app with huggingface spaces.
+
+It doesn't support the pages feature of streamlit yet.
+"""
+import importlib
+from pathlib import Path
+import sys
+
+import streamlit as st
+
+
+def render_main():
+    RIFFUSION_PATH = Path(__file__).parent / "riffusion"
+    sys.path.append(str(RIFFUSION_PATH))
+
+    st.set_page_config(layout="wide", page_icon="🎸")
+
+    # Disable the rest of the setting
+    st.set_page_config = lambda **kwargs: None
+
+    # Find all pages in the riffusion directory
+    pages = sorted(
+        p.name[:-3] for p in (RIFFUSION_PATH / "riffusion" / "streamlit" / "pages").glob("*.py")
+    )
+
+    # Add the pages to the sidebar
+    page = st.sidebar.selectbox("Page", pages, index=pages.index("text_to_audio"))
+    assert page is not None
+
+    module = importlib.import_module(f"riffusion.streamlit.pages.{page}")
+    render_func = getattr(module, f"render_{page}")
+    render_func()
+
+
+render_main()

packages.txt

@@ -0,0 +1,3 @@
+ffmpeg
+libsndfile1
+

requirements.txt

@@ -0,0 +1,17 @@
+accelerate
+argh
+dacite
+diffusers
+flask
+flask_cors
+numpy
+pillow
+plotly
+pydub
+scipy
+soundfile
+sox
+streamlit
+torch
+torchaudio
+transformers

riffusion

@@ -0,0 +1 @@
+Subproject commit 6b02e98e35cd2e4f3a5777fd287ede7be40a8976