Initial commit

README.md

@@ -1,12 +1,14 @@
 ---
 title: GROTTS
-emoji: 💻
-colorFrom: green
+emoji: 🏢
+colorFrom: gray
 colorTo: red
 sdk: gradio
-sdk_version: 4.8.0
+sdk_version: 2.9.1
 app_file: app.py
 pinned: false
+models:
+- bartelds/grotts_vits
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Gronings TTS system (VITS) for three variants: Hoogelaandsters, Oldambsters, and Westerkwartiers.

app.py

@@ -0,0 +1,46 @@
+import gradio as gr
+import time
+import urllib.request
+from pathlib import Path
+import os
+import torch
+import scipy.io.wavfile
+from espnet2.bin.tts_inference import Text2Speech
+from espnet2.utils.types import str_or_none
+
+
+gos_text2speech = Text2Speech.from_pretrained(
+    model_tag="bartelds/grotts_vits",
+    vocoder_tag="none",
+    device="cpu",
+    speed_control_alpha=1.0,
+    noise_scale=1.0,
+    noise_scale_dur=1.0
+)
+
+def inference(text,lang):
+  with torch.no_grad():
+      if lang == "Hoogelaandsters":
+          wav = gos_text2speech(text, sids=np.array([1]))["wav"]
+          scipy.io.wavfile.write("out.wav", gos_text2speech.fs , wav.view(-1).cpu().numpy())
+      if lang == "Oldambsters":
+          wav = gos_text2speech(text, sids=np.array([2]))["wav"]
+          scipy.io.wavfile.write("out.wav", nld_text2speech.fs , wav.view(-1).cpu().numpy())
+      if lang == "Westerkwartiers":
+         wav = gos_text2speech(text, sids=np.array([3]))["wav"]
+         scipy.io.wavfile.write("out.wav", eng_text2speech.fs , wav.view(-1).cpu().numpy())
+
+  return  "out.wav", "out.wav"
+
+title = "GroTTS"
+examples = [
+  ['Ze gingen mit klas noar waddendiek, over en deur bragel lopen.', 'Hoogelaandsters']
+]
+
+gr.Interface(
+    inference,
+    [gr.inputs.Textbox(label="input text", lines=3), gr.inputs.Radio(choices=["Hoogelaandsters", "Oldambsters", "Westerkertaaiers"], type="value", default="Hoogelaandsters", label="language")], 
+    [gr.outputs.Audio(type="file", label="Output"), gr.outputs.File()],
+    title=title,
+    examples=examples
+    ).launch(enable_queue=True)

packages.txt

@@ -0,0 +1,2 @@
+cmake
+libsndfile1

requirements.txt

@@ -0,0 +1,4 @@
+espnet
+espnet_model_zoo
+scipy
+torch