XzJosh commited on
Commit
2ae9025
1 Parent(s): 9d3080c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ from time import time as ttime
30
  from module.mel_processing import spectrogram_torch
31
  from my_utils import load_audio
32
 
33
- device = "cuda"
34
  tokenizer = AutoTokenizer.from_pretrained(bert_path)
35
  bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
36
  if is_half == True:
 
30
  from module.mel_processing import spectrogram_torch
31
  from my_utils import load_audio
32
 
33
+ device = "cpu"
34
  tokenizer = AutoTokenizer.from_pretrained(bert_path)
35
  bert_model = AutoModelForMaskedLM.from_pretrained(bert_path)
36
  if is_half == True: