deepanway commited on
Commit
3c3c9ca
1 Parent(s): cf3c7a3

Change default device to cuda

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from audioldm.variational_autoencoder import AutoencoderKL
10
  from gradio import Markdown
11
 
12
  class Tango:
13
- def __init__(self, name="declare-lab/tango", device="cpu"):
14
 
15
  path = snapshot_download(repo_id=name)
16
 
 
10
  from gradio import Markdown
11
 
12
  class Tango:
13
+ def __init__(self, name="declare-lab/tango", device="cuda:0"):
14
 
15
  path = snapshot_download(repo_id=name)
16