Manasa1 commited on
Commit
6d5be11
·
verified ·
1 Parent(s): 4a72e80

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ import random
13
  script_generator = pipeline("text-generation", model="gpt2", truncation=True, max_length=100)
14
  image_generator = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16).to("cpu")
15
  # For example, use a specific available model if MusicGen isn't accessible
16
- music_generator = pipeline("text-to-music", model="facebook/musicgen")
17
 
18
 
19
  # Generate Comedy Script
 
13
  script_generator = pipeline("text-generation", model="gpt2", truncation=True, max_length=100)
14
  image_generator = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16).to("cpu")
15
  # For example, use a specific available model if MusicGen isn't accessible
16
+ music_generator = pipeline("text-to-music", model="facebook/musicgen-melody")
17
 
18
 
19
  # Generate Comedy Script