stress.py

import argparse
from tts import TTSModel
import os
import time
#import soundfile as sf


model_path = "/Users/dharis/projects/tts_demo/male_naail"
text = 'ރައްޔިތުންގެ މަޖިލީހުގެ އިދާރާ އިން ކުރިއަށް ގެންދާ "އާސްކް ސްޕީކާ"ގެ މިރޭގެ ޕްރޮގްރާމްގައި ނަޝީދު ވިދާޅުވީ ރާއްޖެ އަށް އެންމެ ރަނގަޅު ވާނީ ބަރުލަމާނީ ނިޒާމު ކަމަށް އަދިވެސް ގަބޫލުފުޅު ކުރައްވާ ކަމަށާއި ނަމަވެސް ރިޔާސީ އިންތިހާބެއް އޮވެއްޖެ ނަމަ އެމްޑީޕީގެ ޕްރައިމަރީގައި އެ މަނިކުފާނު ވާދަކުރައްވާނެ ކަމަށެވެ'


tts = TTSModel(
    os.path.join(model_path, "tts.saved_model"),
    os.path.join(model_path, "vocoder.saved_model")
)


for i in range(0,10):
    print("Generating Audio...")
    start = time.time()
    output_wav = tts(text, speed=1)
    end = time.time()

    passed = end - start
    print(f"generated in {passed}")