Upload 6 files

Dockerfile

@@ -14,7 +14,7 @@ ENV PYTHONUNBUFFERED=1 \
     PIP_DISABLE_PIP_VERSION_CHECK=1 \
     DEBIAN_FRONTEND=noninteractive \
     MODELS_DIR=/app/models \
-    GENIE_DATA_DIR=/app/genie_data
+    GENIE_DATA_DIR=/app/GenieData
 
 # Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -45,7 +45,7 @@ RUN pip install --no-cache-dir genie-tts
 # Create directories
 RUN mkdir -p /app/models/liang/onnx \
     && mkdir -p /app/models/liang/reference \
-    && mkdir -p /app/genie_data \
+    && mkdir -p /app/GenieData \
     && mkdir -p /app/temp
 
 # Download model files
@@ -78,9 +78,6 @@ COPY models/liang/config.json /app/models/liang/config.json
 # Copy application code
 COPY app.py .
 
-# 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

@@ -2,15 +2,42 @@
 """
 Model Conversion Script
 =======================
-This script downloads and converts PyTorch models to ONNX format.
+This script downloads GenieData resources and converts PyTorch models to ONNX format.
 """
 
 import os
 import sys
 
+def download_genie_data():
+    """Download GenieData resources from HuggingFace"""
+    from huggingface_hub import snapshot_download
+    
+    genie_data_dir = os.environ.get("GENIE_DATA_DIR", "./GenieData")
+    
+    if os.path.exists(genie_data_dir) and os.listdir(genie_data_dir):
+        print(f"GenieData already exists at {genie_data_dir}")
+        return
+    
+    print("🚀 Starting download Genie-TTS resources from HuggingFace...")
+    snapshot_download(
+        repo_id="High-Logic/Genie",
+        repo_type="model",
+        allow_patterns="GenieData/*",
+        local_dir=".",
+        local_dir_use_symlinks=False,  # Don't use symlinks in Docker
+    )
+    print("✅ Genie-TTS resources downloaded successfully.")
+
 def main():
-    print("Starting ONNX conversion...")
+    # Set environment variable for GenieData location BEFORE importing genie_tts
+    os.environ["GENIE_DATA_DIR"] = "/app/GenieData"
+    
+    # Step 1: Download GenieData resources
+    print("Step 1: Downloading GenieData resources...")
+    download_genie_data()
     
+    # Step 2: Now import genie_tts (it will check for GenieData)
+    print("Step 2: Starting ONNX conversion...")
     import genie_tts as genie
     
     ckpt_path = os.environ.get("CKPT_PATH", "/app/temp/model.ckpt")