wli3221134 commited on
Commit
24fa17c
·
verified ·
1 Parent(s): 6e8a76c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -27,7 +27,8 @@ checkpoint_path = load_model()
27
  @spaces.GPU
28
  def detect_on_gpu(dataset):
29
  """在 GPU 上进行音频伪造检测"""
30
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
31
  model = Wav2Vec2BERT_Llama().to(device)
32
 
33
  # 加载模型权重
 
27
  @spaces.GPU
28
  def detect_on_gpu(dataset):
29
  """在 GPU 上进行音频伪造检测"""
30
+ # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
31
+ device = 'cpu'
32
  model = Wav2Vec2BERT_Llama().to(device)
33
 
34
  # 加载模型权重