cosyvoice / test_tts_simple.py
Marcos Remar
Add test scripts for CosyVoice 1.0 (300M model)
b65e164
raw
history blame contribute delete
816 Bytes
import sys
import os
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
# Import direto sem dependências complexas
import torch
import torchaudio
import numpy as np
print("Teste simples de TTS")
print("Gerando áudio de teste...")
# Criar um áudio de teste simples
sample_rate = 16000
duration = 3 # segundos
frequency = 440 # Hz (nota Lá)
# Gerar onda senoidal
t = np.linspace(0, duration, int(sample_rate * duration))
waveform = 0.3 * np.sin(2 * np.pi * frequency * t)
# Converter para tensor
audio_tensor = torch.FloatTensor(waveform).unsqueeze(0)
# Salvar o áudio
output_file = "test_audio_simple.wav"
torchaudio.save(output_file, audio_tensor, sample_rate)
print(f"Áudio salvo em: {output_file}")
print(f"Duração: {duration} segundos")
print(f"Taxa de amostragem: {sample_rate} Hz")