Update app.py

app.py

@@ -21,9 +21,18 @@ def generate_video(prompt):
 
 # Функция генерации музыки
 def generate_music(prompt):
-    inputs = music_processor(prompt, return_tensors="pt")
-    audio_output = music_model.generate(**inputs)
-    return audio_output
+    inputs = music_processor(text=[prompt], return_tensors="pt")  # Исправленный вызов
+    audio_output = music_model.generate(**inputs)  # Генерация аудио
+    
+    # Декодируем аудио
+    audio_array = audio_output.detach().cpu().numpy().squeeze()
+    
+    # Сохранение в файл
+    audio_path = "generated_audio.wav"
+    sf.write(audio_path, audio_array, samplerate=16000)  # Запись в WAV
+    
+    return audio_path
+    
 
 # Интерфейс Gradio
 with gr.Blocks() as demo: