Manasa1 commited on
Commit
14eb832
·
verified ·
1 Parent(s): 6ab2c25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -45,14 +45,16 @@ def create_images_from_script(script):
45
  image_paths.append(img_path)
46
  return image_paths
47
 
48
- # Generate Fun Music Track using Groq API
49
  def generate_fun_music(prompt, output_music_file="fun_music.wav"):
50
- response = client.music.generate(prompt=prompt)
 
 
51
  audio_data = response["audio"]
52
  sampling_rate = response["sampling_rate"]
53
  scipy.io.wavfile.write(output_music_file, rate=sampling_rate, data=audio_data)
54
  return output_music_file
55
 
 
56
  # Create Video from Generated Images
57
  def generate_text_video(script):
58
  image_paths = create_images_from_script(script)
 
45
  image_paths.append(img_path)
46
  return image_paths
47
 
 
48
  def generate_fun_music(prompt, output_music_file="fun_music.wav"):
49
+ # Use an open-source music generation model
50
+ music_generator = pipeline("text-to-audio", model="facebook/musicgen-small", device="cpu")
51
+ response = music_generator(prompt)
52
  audio_data = response["audio"]
53
  sampling_rate = response["sampling_rate"]
54
  scipy.io.wavfile.write(output_music_file, rate=sampling_rate, data=audio_data)
55
  return output_music_file
56
 
57
+
58
  # Create Video from Generated Images
59
  def generate_text_video(script):
60
  image_paths = create_images_from_script(script)