Spaces:
Running
on
Zero
Running
on
Zero
wli3221134
commited on
Update app.py
Browse files
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 |
# 加载模型权重
|