Update app.py

app.py

@@ -38,7 +38,7 @@ def detect_on_gpu(dataset):
     print(f"正在加载模型权重: {checkpoint_path}")
     checkpoint = torch.load(checkpoint_path, map_location=device)
     model_state_dict = checkpoint['model_state_dict']
-    threshold = 0.9996
+    threshold = 0.8
     print(f"检测阈值设置为: {threshold}")
 
     # 处理模型状态字典的 key
@@ -126,10 +126,12 @@ def audio_deepfake_detection(query_audio_path):
     
     # 调用 GPU 检测函数
     result = detect_on_gpu(audio_dataset)
+    is_fake = "是/Yes" if result["is_fake"] else "否/No"
+    confidence = f"{100*result['confidence']:.2f}%" if result["is_fake"] else f"{100*(1-result['confidence']):.2f}%"
     
     return {
-        "Is AI Generated": result["is_fake"],
-        "Confidence": f"{100*result['confidence']:.2f}%"
+        "是否为AI生成/Is AI Generated": is_fake,
+        "检测可信度/Confidence": confidence
     }
 
 # Gradio 界面