Update app.py

app.py

@@ -2,21 +2,21 @@ import gradio as gr
 import librosa
 import numpy as np
 import torch
-from transformers import AutoProcessor, AutoModelForAudioClassification
+from transformers import AutoFeatureExtractor, AutoModelForAudioClassification
 import tempfile
 import soundfile as sf
 import json
 
 SAMPLE_RATE = 16000
-CHUNK_SIZE = 60  # 模型要求60秒输入
-STEP = 10        # 滑动步长（秒）
+CHUNK_SIZE = 60
+STEP = 10
 MUSIC_THRESHOLD = 0.5
 VOICE_THRESHOLD = 0.3
-MIN_SEG_DURATION = 8  # 最小合并段时长
+MIN_SEG_DURATION = 8
 
-# === 加载固定模型（适用于 60s 音频输入） ===
+# === 修正版 ===
 music_model_id = "AI-Music-Detection/ai_music_detection_large_60s"
-music_processor = AutoProcessor.from_pretrained(music_model_id)
+music_extractor = AutoFeatureExtractor.from_pretrained("MIT/ast-finetuned-audioset-10-10-0.4593")
 music_model = AutoModelForAudioClassification.from_pretrained(music_model_id)
 
 voice_model_id = "superb/hubert-large-superb-sid"