Rooni commited on
Commit
3e4e74a
·
verified ·
1 Parent(s): a16db00

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +35 -0
app.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from huggingface_hub import cached_download, hf_hub_url
3
+ import torch
4
+ from transformers import pipeline
5
+
6
+ model_id = "facebook/audiogen-medium"
7
+ audiogen = pipeline("text-to-audio", model=model_id)
8
+
9
+
10
+ def generate_audio(text, duration):
11
+ audio = audiogen(text, duration=duration)
12
+ return audio.audio
13
+
14
+
15
+ with gr.Blocks() as demo:
16
+ gr.Markdown(
17
+ """
18
+ # Генератор музыки из текста
19
+ """
20
+ )
21
+ with gr.Row():
22
+ text_input = gr.Textbox(label="Введите текст для генерации музыки:")
23
+ duration_slider = gr.Slider(
24
+ minimum=2, maximum=30, value=10, label="Длина аудио (сек.):"
25
+ )
26
+ generate_button = gr.Button("Сгенерировать музыку")
27
+ audio_output = gr.Audio(label="Результат:")
28
+
29
+ generate_button.click(
30
+ fn=generate_audio,
31
+ inputs=[text_input, duration_slider],
32
+ outputs=audio_output,
33
+ )
34
+
35
+ demo.launch()