spidy254 commited on
Commit
0ada2c5
·
verified ·
1 Parent(s): 502ffce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ from diffusers import StableDiffusionPipeline
8
 
9
  # Load AI Art Model (Stable Diffusion)
10
  pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
11
- pipe.to("cuda" if torch.cuda.is_available() else "cpu")
12
 
13
  def process_audio(audio_file):
14
  # Load audio file
 
8
 
9
  # Load AI Art Model (Stable Diffusion)
10
  pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
11
+ pipe.to("cpu")
12
 
13
  def process_audio(audio_file):
14
  # Load audio file