Spaces:
Sleeping
Sleeping
File size: 2,479 Bytes
5efbc82 1b567fa 5efbc82 1b567fa 5efbc82 1b567fa 5efbc82 |
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 |
#!/usr/bin/env python3
"""
Simple test script to verify Kokoro TTS installation and functionality.
"""
import os
# Set basic environment variables
os.environ['NUMBA_DISABLE_JIT'] = '1'
def test_kokoro_import():
"""Test if Kokoro can be imported"""
try:
from kokoro import KPipeline
import soundfile as sf
import torch
print("✅ All required packages imported successfully!")
return True
except ImportError as e:
print(f"❌ Import error: {e}")
return False
def test_kokoro_pipeline():
"""Test if Kokoro pipeline can be initialized"""
try:
from kokoro import KPipeline
pipeline = KPipeline(lang_code='a')
print("✅ Kokoro pipeline initialized successfully!")
return True
except Exception as e:
print(f"❌ Pipeline initialization error: {e}")
return False
def test_kokoro_generation():
"""Test if Kokoro can generate speech"""
try:
from kokoro import KPipeline
import soundfile as sf
pipeline = KPipeline(lang_code='a')
text = "Hello, this is a test of Kokoro TTS."
generator = pipeline(text, voice='af_heart')
for i, (gs, ps, audio) in enumerate(generator):
print(f"✅ Generated audio segment {i}: gs={gs}, ps={ps}")
# Save test audio
sf.write('test_kokoro.wav', audio, 24000)
print("✅ Test audio saved as 'test_kokoro.wav'")
break # Just test the first segment
return True
except Exception as e:
print(f"❌ Speech generation error: {e}")
return False
def main():
"""Run all tests"""
print("🎤 Testing Kokoro TTS Installation")
print("=" * 40)
tests = [
("Import Test", test_kokoro_import),
("Pipeline Test", test_kokoro_pipeline),
("Generation Test", test_kokoro_generation)
]
passed = 0
total = len(tests)
for test_name, test_func in tests:
print(f"\n🔍 Running {test_name}...")
if test_func():
passed += 1
else:
print(f"❌ {test_name} failed!")
print(f"\n📊 Results: {passed}/{total} tests passed")
if passed == total:
print("🎉 All tests passed! Kokoro TTS is ready to use.")
else:
print("⚠️ Some tests failed. Please check the installation.")
if __name__ == "__main__":
main() |