mirror of
https://github.com/HumanAIGC-Engineering/gradio-webrtc.git
synced 2026-02-05 09:59:22 +08:00
Add first-class support for Cartesia text-to-speech (#298)
* Demo * patient intake * cartesia * Add cartesia * Fix * lint * Move test * Fix * Fix * Fix * Fix
This commit is contained in:
11
test/test_tts.py
Normal file
11
test/test_tts.py
Normal file
@@ -0,0 +1,11 @@
|
||||
import pytest
|
||||
from fastrtc.text_to_speech.tts import get_tts_model
|
||||
|
||||
|
||||
@pytest.mark.parametrize("model", ["kokoro"])
|
||||
def test_tts_long_prompt(model):
|
||||
model = get_tts_model(model=model)
|
||||
prompt = "It may be that this communication will be considered as a madman's freak but at any rate it must be admitted that in its clearness and frankness it left nothing to be desired The serious part of it was that the Federal Government had undertaken to treat a sale by auction as a valid concession of these undiscovered territories Opinions on the matter were many Some readers saw in it only one of those prodigious outbursts of American humbug which would exceed the limits of puffism if the depths of human credulity were not unfathomable"
|
||||
|
||||
for i, chunk in enumerate(model.stream_tts_sync(prompt)):
|
||||
print(f"Chunk {i}: {chunk[1].shape}")
|
||||
@@ -39,6 +39,7 @@ class MinimalTestStream(WebRTCConnectionMixin):
|
||||
)
|
||||
self.time_limit = time_limit
|
||||
self.allow_extra_tracks = allow_extra_tracks
|
||||
self.server_rtc_configuration = None
|
||||
|
||||
def mount(self, app: FastAPI, path: str = ""):
|
||||
from fastapi import APIRouter
|
||||
|
||||
Reference in New Issue
Block a user