feat: change model name

app/export.py

@@ -1,7 +1,6 @@
 from glob import glob
 import os
 import shutil
-import tempfile
 import gradio as gr
 from infer.lib.train.process_ckpt import extract_small_model
 from app.train import train_index
@@ -15,8 +14,7 @@ def download_weight(exp_dir: str) -> str:
     latest_checkpoint = max(checkpoints, key=os.path.getctime)
     print(f"Latest checkpoint: {latest_checkpoint}")
 
-    name = os.path.basename(exp_dir)
-    out = os.path.join(exp_dir, f"{name}.pth")
+    out = os.path.join(exp_dir, f"model.pth")
     extract_small_model(
         latest_checkpoint, out, "40k", True, "Model trained by ZeroGPU.", "v2"
     )

app/infer.py

@@ -15,8 +15,7 @@ from model import device
 def infer(
     exp_dir: str, original_audio: str, f0add: int, index_rate: float, protect: float
 ) -> Tuple[int, np.ndarray]:
-    name = os.path.basename(exp_dir)
-    model = os.path.join(exp_dir, f"{name}.pth")
+    model = os.path.join(exp_dir, "model.pth")
     if not os.path.exists(model):
         raise gr.Error("Model not found")