Update app.py
Browse files
app.py
CHANGED
|
@@ -5,34 +5,28 @@ import gradio as gr
|
|
| 5 |
import soundfile as sf
|
| 6 |
from huggingface_hub import InferenceClient
|
| 7 |
|
| 8 |
-
# HF_TOKEN aus Secret
|
| 9 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 10 |
if not HF_TOKEN:
|
| 11 |
raise RuntimeError(
|
| 12 |
"HF_TOKEN fehlt! Bitte in den Space Settings → Secrets eintragen."
|
| 13 |
)
|
| 14 |
|
| 15 |
-
# MusicGen Client
|
| 16 |
client = InferenceClient(model="facebook/musicgen-small", token=HF_TOKEN)
|
| 17 |
|
| 18 |
-
# Musik generieren
|
| 19 |
def generate_music(prompt: str):
|
| 20 |
if not prompt.strip():
|
| 21 |
return None
|
| 22 |
audio_bytes = client.text_to_audio(prompt)
|
| 23 |
audio_data, samplerate = sf.read(io.BytesIO(audio_bytes))
|
| 24 |
-
# Konvertiere in float32, damit Gradio stabil ist
|
| 25 |
audio_data = np.array(audio_data, dtype=np.float32)
|
| 26 |
return samplerate, audio_data
|
| 27 |
|
| 28 |
-
# Gradio Interface
|
| 29 |
demo = gr.Interface(
|
| 30 |
fn=generate_music,
|
| 31 |
inputs=gr.Textbox(label="Beschreibe deine Musik", placeholder="z.B. Lo-fi chill"),
|
| 32 |
outputs=gr.Audio(label="Generierte Musik", type="numpy"),
|
| 33 |
title="🎶 KI Musik Generator",
|
| 34 |
-
description="Musik wird mit MusicGen generiert. Dauer ca. 30–60 Sekunden (GPU empfohlen)."
|
| 35 |
-
allow_flagging="never"
|
| 36 |
)
|
| 37 |
|
| 38 |
if __name__ == "__main__":
|
|
|
|
| 5 |
import soundfile as sf
|
| 6 |
from huggingface_hub import InferenceClient
|
| 7 |
|
|
|
|
| 8 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 9 |
if not HF_TOKEN:
|
| 10 |
raise RuntimeError(
|
| 11 |
"HF_TOKEN fehlt! Bitte in den Space Settings → Secrets eintragen."
|
| 12 |
)
|
| 13 |
|
|
|
|
| 14 |
client = InferenceClient(model="facebook/musicgen-small", token=HF_TOKEN)
|
| 15 |
|
|
|
|
| 16 |
def generate_music(prompt: str):
|
| 17 |
if not prompt.strip():
|
| 18 |
return None
|
| 19 |
audio_bytes = client.text_to_audio(prompt)
|
| 20 |
audio_data, samplerate = sf.read(io.BytesIO(audio_bytes))
|
|
|
|
| 21 |
audio_data = np.array(audio_data, dtype=np.float32)
|
| 22 |
return samplerate, audio_data
|
| 23 |
|
|
|
|
| 24 |
demo = gr.Interface(
|
| 25 |
fn=generate_music,
|
| 26 |
inputs=gr.Textbox(label="Beschreibe deine Musik", placeholder="z.B. Lo-fi chill"),
|
| 27 |
outputs=gr.Audio(label="Generierte Musik", type="numpy"),
|
| 28 |
title="🎶 KI Musik Generator",
|
| 29 |
+
description="Musik wird mit MusicGen generiert. Dauer ca. 30–60 Sekunden (GPU empfohlen)."
|
|
|
|
| 30 |
)
|
| 31 |
|
| 32 |
if __name__ == "__main__":
|