Update app.py

app.py

@@ -13,7 +13,8 @@ import random
 script_generator = pipeline("text-generation", model="gpt2", truncation=True, max_length=100)
 image_generator = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16).to("cpu")
 # For example, use a specific available model if MusicGen isn't accessible
-music_generator = pipeline("text-to-music", model="facebook/musicgen-melody")
+music_generator = pipeline("text-to-audio", model="facebook/musicgen-melody", device="cpu")
+
 
 
 # Generate Comedy Script
@@ -28,6 +29,7 @@ def text_to_speech(script):
     tts.save(audio_file)
     return audio_file
 
+
 # Create Images Using Stable Diffusion
 def create_images_from_script(script):
     lines = script.split('. ')