Update app.py

app.py

@@ -1,28 +1,33 @@
 import gradio as gr
-from transformers import AutoModel, AutoTokenizer  # 替换AutoModelForTextToSpeech为AutoModel
+from transformers import AutoModel, AutoTokenizer
 import soundfile as sf
 import torch
 import os
 
-# 加载模型和Tokenizer（自动下载SoulX模型）
+# 加载模型和Tokenizer（修复参数+移除device_map）
 model_name = "Soul-AILab/SoulX-Podcast-1.7B"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModel.from_pretrained(  # 这里用AutoModel替代AutoModelForTextToSpeech
+model = AutoModel.from_pretrained(
     model_name,
-    torch_dtype=torch.float16,
-    device_map="auto"
+    dtype=torch.float16,  # 替换 deprecated 的 torch_dtype
+    # 移除 device_map="auto"，改用手动分配设备（兼容无accelerate环境）
 )
+# 手动将模型移到GPU（无GPU自动用CPU）
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to(device)
 
-# 语音生成函数（逻辑不变）
+# 语音生成函数（补充设备适配）
 def generate_speech(text):
     if not text.strip():
         return None, "错误：请输入有效文本！"
     
-    inputs = tokenizer(text, return_tensors="pt").to(model.device)
+    # 文本编码并移到对应设备
+    inputs = tokenizer(text, return_tensors="pt").to(device)
     
     with torch.no_grad():
         audio_output = model.generate(**inputs)
     
+    # 保存音频
     output_path = "output.wav"
     sf.write(output_path, audio_output[0].cpu().numpy(), samplerate=24000)