Renday commited on
Commit
8da966f
·
verified ·
1 Parent(s): cf18642

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -7
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__":