Update app.py

app.py

@@ -16,25 +16,43 @@ torch.nn.Module.cuda = no_op
 print("💉 CUDA 已屏蔽")
 
 # ==========================================
-# 2. 核心补丁：补全 CPU 缺失文件
+# 2. 饱和式救援：下载全套缺失零件
 # ==========================================
-print("🏗️ 启动依赖补全计划...")
+print("🚚 启动饱和式空投，正在重建 CPU 环境...")
 
+BASE_URL = "https://raw.githubusercontent.com/RVC-Boss/GPT-SoVITS/main/GPT_SoVITS"
+
+# 定义所有需要修复的文件清单
 FILES_TO_PATCH = [
+    # 核心：把 CPU 模型代码写入 GPU 文件
     {
-        "url": "https://raw.githubusercontent.com/RVC-Boss/GPT-SoVITS/main/GPT_SoVITS/AR/models/t2s_model.py",
-        "path": "AR/models/t2s_model_flash_attn.py", # 覆盖目标
-        "desc": "模型主文件 (CPU替换GPU)"
+        "url": f"{BASE_URL}/AR/models/t2s_model.py",
+        "path": "AR/models/t2s_model_flash_attn.py",
     },
+    # 依赖 1: Utils
     {
-        "url": "https://raw.githubusercontent.com/RVC-Boss/GPT-SoVITS/main/GPT_SoVITS/AR/models/utils.py",
+        "url": f"{BASE_URL}/AR/models/utils.py",
         "path": "AR/models/utils.py",
-        "desc": "工具库 (Utils)"
     },
+    # 依赖 2: Embedding
     {
-        "url": "https://raw.githubusercontent.com/RVC-Boss/GPT-SoVITS/main/GPT_SoVITS/AR/modules/embedding.py",
+        "url": f"{BASE_URL}/AR/modules/embedding.py",
         "path": "AR/modules/embedding.py",
-        "desc": "嵌入层 (Embedding)"
+    },
+    # 依赖 3: Transformer (你刚刚报错缺这个)
+    {
+        "url": f"{BASE_URL}/AR/modules/transformer.py",
+        "path": "AR/modules/transformer.py",
+    },
+    # 依赖 4: Attention (为了防止还没报错就先补上)
+    {
+        "url": f"{BASE_URL}/AR/modules/attention.py",
+        "path": "AR/modules/attention.py",
+    },
+    # 依赖 5: Commons (保险起见)
+    {
+        "url": f"{BASE_URL}/AR/modules/commons.py",
+        "path": "AR/modules/commons.py",
     }
 ]
 
@@ -44,26 +62,23 @@ for item in FILES_TO_PATCH:
         dir_name = os.path.dirname(item["path"])
         if not os.path.exists(dir_name):
             os.makedirs(dir_name, exist_ok=True)
-            print(f"📁 创建目录: {dir_name}")
-            
-            # 补一个 __init__.py 确保可导入
+            # 补 init
             with open(os.path.join(dir_name, "__init__.py"), "w") as f: f.write("")
 
-        # 2. 下载文件
-        print(f"⬇️ 下载 {item['desc']} ...")
+        # 2. 下载
+        print(f"⬇️ 下载补丁: {os.path.basename(item['path'])} ...")
         resp = requests.get(item["url"], timeout=10)
         if resp.status_code == 200:
-            # 3. 写入文件
             with open(item["path"], "w", encoding="utf-8") as f:
                 f.write(resp.text)
-            print(f"✅ 写入成功: {item['path']}")
+            print(f"✅ 修复成功: {item['path']}")
         else:
-            print(f"❌ 下载失败 {item['url']}: {resp.status_code}")
+            print(f"❌ 下载失败 ({resp.status_code}): {item['url']}")
             
     except Exception as e:
-        print(f"⚠️ 处理 {item['path']} 时出错: {e}")
+        print(f"⚠️ 文件处理错误: {e}")
 
-# 补充根目录 AR 的 __init__.py
+# 补充根目录 init
 if os.path.exists("AR") and not os.path.exists("AR/__init__.py"):
     with open("AR/__init__.py", "w") as f: f.write("")
 
@@ -108,7 +123,7 @@ try:
             core.change_gpt_weights(gpt_path=gpt_path)
         if hasattr(core, "change_sovits_weights"):
             core.change_sovits_weights(sovits_path=sovits_path)
-        print(f"🎉 模型加载成功！(CPU Full Patch)")
+        print(f"🎉 模型加载成功！(CPU Rebuilt)")
     else:
         print("❌ 未找到模型文件")
 except Exception as e:
@@ -160,16 +175,13 @@ def run_predict(text):
 # 7. 界面
 # ==========================================
 with gr.Blocks() as app:
-    gr.Markdown(f"### GPT-SoVITS V2 (CPU Full Repair)")
-    
+    gr.Markdown(f"### GPT-SoVITS V2 (Full Repair)")
     with gr.Row():
-        inp = gr.Textbox(label="文本", value="这下真的真的真的要成功了。")
+        inp = gr.Textbox(label="文本", value="所有零件都补齐了，这次一定行。")
         btn = gr.Button("生成")
-    
     with gr.Row():
         out = gr.Audio(label="结果")
         log = gr.Textbox(label="日志")
-        
     btn.click(run_predict, [inp], [out, log], api_name="predict")
 
 if __name__ == "__main__":