add tts model

app.py

@@ -75,6 +75,38 @@ from langchain.schema import (
     LLMResult
     )
 import time
+from datasets import load_dataset
+
+from transformers import pipeline
+
+
+ds = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
+sample = ds[0]["audio"]
+
+def speech_to_text_loc(audio):
+    device = "cpu"
+    pipe = pipeline(
+    "automatic-speech-recognition",
+    model="openai/whisper-small",
+    chunk_length_s=30,
+    device=device,
+    )
+    text = pipe(audio.copy(), batch_size=2)["text"]
+    return text
+
+print("voice to text loc: ", speech_to_text_loc(sample))
+
+def text_to_speech_loc(audio):
+    device = "cpu"
+    pipe = pipeline(
+    "text-to-speech",
+    model="microsoft/speecht5_tts",
+    device=device,
+    )
+    text = pipe(audio.copy(), batch_size=2)["text"]
+    return text
+
+print("text to speech: ", text_to_speech_loc("Good morning."))
 
 class GPTRemote(LLM):
     n: int