Spaces:
Configuration error
Configuration error
File size: 3,549 Bytes
44742c4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 |
#!/usr/bin/env python3
import os
import sys
import subprocess
import shutil
from pathlib import Path
# Configuration
REPO_URL = "https://github.com/hpbyte/myanmar-tts.git"
REPO_DIR = "myanmar-tts"
MODEL_DIR = "trained_model"
def setup_repository():
"""Clone and set up the Myanmar TTS repository."""
print(f"Setting up Myanmar TTS repository...")
# Clone the repository if it doesn't exist
if not os.path.exists(REPO_DIR):
print(f"Cloning repository from {REPO_URL}...")
subprocess.run(["git", "clone", REPO_URL], check=True)
print("Repository cloned successfully!")
else:
print(f"Repository already exists at {REPO_DIR}")
# Create the model directory if it doesn't exist
if not os.path.exists(MODEL_DIR):
os.makedirs(MODEL_DIR)
print(f"Created model directory at {MODEL_DIR}")
else:
print(f"Model directory already exists at {MODEL_DIR}")
# Create placeholder files with instructions if they don't exist
checkpoint_path = os.path.join(MODEL_DIR, "checkpoint_latest.pth.tar")
config_path = os.path.join(MODEL_DIR, "hparams.yml")
if not os.path.exists(checkpoint_path):
with open(checkpoint_path, 'w') as f:
f.write("This is a placeholder file. Replace with the actual model checkpoint.")
print(f"Created placeholder file at {checkpoint_path}")
if not os.path.exists(config_path):
with open(config_path, 'w') as f:
f.write("# This is a placeholder file. Replace with the actual hparams.yml file.")
print(f"Created placeholder file at {config_path}")
# Make sure the repository packages are installed
try:
# Create a minimal setup.py in the repository if it doesn't exist
setup_path = os.path.join(REPO_DIR, "setup.py")
if not os.path.exists(setup_path):
with open(setup_path, 'w') as f:
f.write("""
from setuptools import setup, find_packages
setup(
name="myanmar_tts",
version="0.1.0",
packages=find_packages(),
install_requires=[
"inflect",
"unidecode",
"torch",
"numpy",
"scipy",
],
)
""")
print("Created setup.py file in repository")
# Install the package in development mode
print("Installing Myanmar TTS package...")
subprocess.run([sys.executable, "-m", "pip", "install", "-e", REPO_DIR], check=True)
print("Myanmar TTS package installed successfully!")
# Create __init__.py files to make imports work
for subdir in ["text", "utils"]:
init_path = os.path.join(REPO_DIR, subdir, "__init__.py")
if not os.path.exists(init_path):
os.makedirs(os.path.dirname(init_path), exist_ok=True)
with open(init_path, 'w') as f:
f.write("# Auto-generated __init__.py file")
print(f"Created {init_path}")
except Exception as e:
print(f"Error during package installation: {str(e)}")
print("Continuing with setup...")
print("Repository setup complete!")
print("\nIMPORTANT: You need to upload the following files to the 'trained_model' directory:")
print("1. checkpoint_latest.pth.tar - The model checkpoint file")
print("2. hparams.yml - The hyperparameters configuration file")
print("\nThese files can be obtained from the original repository or by training the model.")
if __name__ == "__main__":
setup_repository() |