Adding possibility to read the input text from stdin in the CLI (#110)
This commit is contained in:
@@ -1,6 +1,7 @@
|
|||||||
import io
|
import io
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
|
import sys
|
||||||
import tempfile
|
import tempfile
|
||||||
import threading
|
import threading
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
@@ -243,6 +244,13 @@ def generate(
|
|||||||
|
|
||||||
log_level = logging.ERROR if quiet else logging.INFO
|
log_level = logging.ERROR if quiet else logging.INFO
|
||||||
with enable_logging("pocket_tts", log_level):
|
with enable_logging("pocket_tts", log_level):
|
||||||
|
if text == "-":
|
||||||
|
# Read text from stdin
|
||||||
|
text = sys.stdin.read()
|
||||||
|
|
||||||
|
if not text.strip():
|
||||||
|
logger.error("No input received from stdin.")
|
||||||
|
raise typer.Exit(code=1)
|
||||||
tts_model = TTSModel.load_model(
|
tts_model = TTSModel.load_model(
|
||||||
config, temperature, lsd_decode_steps, noise_clamp, eos_threshold
|
config, temperature, lsd_decode_steps, noise_clamp, eos_threshold
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user