Update app.py

app.py

@@ -15,8 +15,8 @@ video_pipe = StableVideoDiffusionPipeline.from_pretrained(
 ).to(device)
 
 # Загружаем модель для генерации музыки (MusicGen)
-music_model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
-music_processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")
+music_model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-tiny")
+music_processor = MusicgenProcessor.from_pretrained("facebook/musicgen-tiny")
 
 # Функция генерации видео
 def generate_video(prompt):
@@ -33,7 +33,7 @@ def generate_video(prompt):
 # Функция генерации музыки
 def generate_music(prompt):
     inputs = music_processor(text=[prompt], return_tensors="pt")  # Правильная передача текста
-    audio_output = music_model.generate(**inputs)  # Генерация аудио
+    audio_output = music_model.generate(**inputs, max_new_tokens=512)  # Генерация аудио
 
     # Декодирование аудиовыхода
     audio_array = audio_output[0].cpu().detach().numpy().squeeze()