Update app.py

app.py

@@ -8,13 +8,15 @@ from huggingface_hub import hf_hub_download
 
 # 初始化设备
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+print('device:', device)
 
 # 初始化模型
 def load_model():
     model = Wav2Vec2BERT_Llama().to(device)
     checkpoint_path = hf_hub_download(
         repo_id="amphion/deepfake_detection", 
-        filename="checkpoints_wav2vec2bert_ft_llama_labels_ASVspoof2019_RandomPrompts_6/model_checkpoint.pth"
+        filename="checkpoints_wav2vec2bert_ft_llama_labels_ASVspoof2019_RandomPrompts_6/model_checkpoint.pth",
+        repo_type="model"
     )
     # checkpoint_path = "ckpt/model_checkpoint.pth"
     if os.path.exists(checkpoint_path):