Upload 6 files

Dockerfile

@@ -59,18 +59,12 @@ RUN echo "Downloading model files..." && \
     "https://22333misaka-openlist.hf.space/d/od/ref_shantianliang_1.wav" && \
     echo "Download complete!"
 
+# Copy conversion script
+COPY convert_model.py .
+
 # Convert PyTorch models to ONNX
 RUN echo "Converting models to ONNX format..." && \
-    python -c "
-import genie_tts as genie
-print('Starting ONNX conversion...')
-genie.convert_to_onnx(
-    torch_ckpt_path='/app/temp/model.ckpt',
-    torch_pth_path='/app/temp/model.pth',
-    output_dir='/app/models/liang/onnx'
-)
-print('ONNX conversion complete!')
-" && \
+    python convert_model.py && \
     echo "Conversion complete!"
 
 # Clean up temporary files and torch to reduce image size
@@ -78,18 +72,14 @@ RUN rm -rf /app/temp && \
     pip uninstall -y torch && \
     rm -rf /root/.cache/pip
 
-# Create model configuration
-RUN echo '{\n\
-    "reference_audio": "reference/audio.wav",\n\
-    "reference_text": "这是一条参考音频，将此音频拖入参考内，再添加文本，即可合成音色",\n\
-    "language": "Chinese"\n\
-}' > /app/models/liang/config.json
+# Copy model configuration
+COPY models/liang/config.json /app/models/liang/config.json
 
 # Copy application code
 COPY app.py .
 
-# Download Genie base data
-RUN python -c "import genie_tts; genie_tts.download_genie_data()"
+# Download Genie base data (uses a script to avoid inline Python issues)
+RUN python -c 'import genie_tts; genie_tts.download_genie_data()'
 
 # Expose port (Hugging Face Spaces uses 7860)
 EXPOSE 7860

convert_model.py

@@ -0,0 +1,47 @@
+#!/usr/bin/env python3
+"""
+Model Conversion Script
+=======================
+This script downloads and converts PyTorch models to ONNX format.
+"""
+
+import os
+import sys
+
+def main():
+    print("Starting ONNX conversion...")
+    
+    import genie_tts as genie
+    
+    ckpt_path = os.environ.get("CKPT_PATH", "/app/temp/model.ckpt")
+    pth_path = os.environ.get("PTH_PATH", "/app/temp/model.pth")
+    output_dir = os.environ.get("OUTPUT_DIR", "/app/models/liang/onnx")
+    
+    print(f"CKPT path: {ckpt_path}")
+    print(f"PTH path: {pth_path}")
+    print(f"Output dir: {output_dir}")
+    
+    # Check if files exist
+    if not os.path.exists(ckpt_path):
+        print(f"Error: CKPT file not found: {ckpt_path}")
+        sys.exit(1)
+    
+    if not os.path.exists(pth_path):
+        print(f"Error: PTH file not found: {pth_path}")
+        sys.exit(1)
+    
+    # Create output directory
+    os.makedirs(output_dir, exist_ok=True)
+    
+    # Convert model
+    genie.convert_to_onnx(
+        torch_ckpt_path=ckpt_path,
+        torch_pth_path=pth_path,
+        output_dir=output_dir
+    )
+    
+    print("ONNX conversion complete!")
+
+
+if __name__ == "__main__":
+    main()