Hugging Face's logo

Spaces:
piton13332
/
dance-diffusion
Running

App Files Community
piton13332 commited on
Commit
ebb39eb
·
verified ·
1 Parent(s): 18f588c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import scipy.io.wavfile
5
 
6
  def load_model(model_id):
7
  pipeline = DiffusionPipeline.from_pretrained(model_id)
8
- pipeline = pipeline.to("cuda")
9
  return pipeline
10
 
11
  def denoise(length_sec,model):
 
5
 
6
  def load_model(model_id):
7
  pipeline = DiffusionPipeline.from_pretrained(model_id)
8
+ pipeline = pipeline.to("cpu")
9
  return pipeline
10
 
11
  def denoise(length_sec,model):