Manasa1 commited on
Commit
394d306
·
verified ·
1 Parent(s): 3f22c4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -25,7 +25,8 @@ music_generator = pipeline("text-to-audio", model="facebook/musicgen-small", dev
25
 
26
  # Use Fluently Anime (Stable Diffusion) for anime image generation
27
  model_id = "fluently/Fluently-anime"
28
- anime_image_generator = StableDiffusionPipeline.from_pretrained(model_id).to("cuda")
 
29
 
30
  # Generate Comedy Script using DistilGPT-2
31
  def generate_comedy_script(prompt):
 
25
 
26
  # Use Fluently Anime (Stable Diffusion) for anime image generation
27
  model_id = "fluently/Fluently-anime"
28
+ anime_image_generator = StableDiffusionPipeline.from_pretrained(model_id).to("cpu")
29
+
30
 
31
  # Generate Comedy Script using DistilGPT-2
32
  def generate_comedy_script(prompt):