AiCoderv2 commited on
Commit
88e2c72
·
verified ·
1 Parent(s): 91fd009

Update app.py from anycoder

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -91,8 +91,9 @@ def generate_music(prompt, duration, temperature, top_k):
91
  top_k=top_k
92
  )
93
 
94
- # Get sampling rate from model config
95
- sampling_rate = model.config.audio_encoder.sample_rate
 
96
 
97
  # Convert audio tensor to numpy array
98
  # MusicGen outputs audio in stereo (2 channels)
 
91
  top_k=top_k
92
  )
93
 
94
+ # Get sampling rate from processor's audio encoder config
95
+ # MusicGen uses Encodec for audio processing
96
+ sampling_rate = processor.feature_extractor.sampling_rate
97
 
98
  # Convert audio tensor to numpy array
99
  # MusicGen outputs audio in stereo (2 channels)