Nikolaygod777 commited on
Commit
a5e6912
·
verified ·
1 Parent(s): 456ba3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ video_pipe = StableVideoDiffusionPipeline.from_pretrained(
15
  ).to(device)
16
 
17
  # Загружаем модель для генерации музыки (MusicGen)
18
- music_model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-tiny")
19
- music_processor = MusicgenProcessor.from_pretrained("facebook/musicgen-tiny")
20
 
21
  # Функция генерации видео
22
  def generate_video(prompt):
 
15
  ).to(device)
16
 
17
  # Загружаем модель для генерации музыки (MusicGen)
18
+ music_model = MusicgenForConditionalGeneration.from_pretrained("facebook/musicgen-small")
19
+ music_processor = MusicgenProcessor.from_pretrained("facebook/musicgen-small")
20
 
21
  # Функция генерации видео
22
  def generate_video(prompt):