Manasa1 commited on
Commit
931a180
·
verified ·
1 Parent(s): 7ae35c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -12,7 +12,9 @@ import random
12
  # Load and Initialize Models
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
- music_generator = pipeline("text-to-music", model="HuggingFace/MusicGen") # Replace with actual model if available
 
 
16
 
17
  # Generate Comedy Script
18
  def generate_comedy_script(prompt):
 
12
  # Load and Initialize Models
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
20
  def generate_comedy_script(prompt):