Change format for voice selection

Closes #2

app.py

@@ -15,6 +15,8 @@ except ImportError:
 
 import torch
 import torchaudio
+import gradio as gr
+
 
 from huggingface_hub import hf_hub_download
 
@@ -24,7 +26,6 @@ from data import TextProcessor
 from common import update_params
 from torch_env import device
 
-import gradio as gr
 
 
 # Vocoder
@@ -184,44 +185,31 @@ tech_libraries = f"""
 - librosa: {version("librosa")}
 """.strip()
 
-
-class VoiceOption(Enum):
-    Tetiana = "Tetiana (female) 👩"
-    Mykyta = "Mykyta (male) 👨"
-    Lada = "Lada (female) 👩"
-
-
-voice_mapping = {
-    VoiceOption.Tetiana.value: "tetiana",
-    VoiceOption.Mykyta.value: "mykyta",
-    VoiceOption.Lada.value: "lada",
+voices = {
+    "lada": 0,
+    "mykyta": 1,
+    "tetiana": 2,
 }
 
 
 examples = [
     [
         "Прокинувся ґазда вранці. Пішов, вичистив з-під коня, вичистив з-під бика, вичистив з-під овечок, вибрав молодняк, відніс його набік.",
-        VoiceOption.Mykyta.value,
+        "Mykyta",
     ],
     [
         "Пішов взяв сіна, дав корові. Пішов взяв сіна, дав бикові. Ячміню коняці насипав. Зайшов почистив корову, зайшов почистив бика, зайшов почистив коня, за яйця його мацнув.",
-        VoiceOption.Lada.value,
+        "Lada",
     ],
     [
         "Кінь ногою здригнув, на хазяїна ласкавим оком подивився. Тоді дядько пішов відкрив курей, гусей, качок, повиносив їм зерна, огірків нарізаних, нагодував. Коли чує – з хати дружина кличе. Зайшов. Дітки повмивані, сидять за столом, всі чекають тата. Взяв він ложку, перехрестив дітей, перехрестив лоба, почали снідати. Поснідали, він дістав пряників, роздав дітям. Діти зібралися, пішли в школу. Дядько вийшов, сів на призьбі, взяв сапку, почав мантачити. Мантачив-мантачив, коли – жінка виходить. Він їй ту сапку дає, ласкаво за сраку вщипнув, жінка до нього лагідно всміхнулася, пішла на город – сапати. Коли – йде пастух і товар кличе в череду. Повідмикав дядько овечок, коровку, бика, коня, все відпустив. Сів попри хати, дістав табАку, відірвав шмат газети, насипав, наслинив собі гарну таку цигарку. Благодать божа – і сонечко вже здійнялося над деревами. Дядько встромив цигарку в рота, дістав сірники, тільки чиркати – коли раптом з хати: Доброе утро! Московское время – шесть часов утра! Витяг дядько цигарку с рота, сплюнув набік, і сам собі каже: Ана маєш. Прокинулись, бляді!",
-        VoiceOption.Tetiana.value,
+        "Tetiana",
     ],
 ]
 
 
 def get_speaker_id(speaker):
-    speaker_ids = {
-        "lada": 0,
-        "mykyta": 1,
-        "tetiana": 2,
-    }
-
-    return torch.LongTensor([speaker_ids[speaker]])
+    return torch.LongTensor([voices[speaker]])
 
 
 def get_text(text):
@@ -234,7 +222,7 @@ def inference(text, voice):
 
     gr.Info("Starting...", duration=0.5)
 
-    speaker = voice_mapping[voice]
+    speaker = voice.lower()
     speaker = speaker_text = speaker_attributes = speaker
 
     n_takes = 1
@@ -350,8 +338,8 @@ with demo:
             )
             voice = gr.Radio(
                 label="Voice",
-                choices=[option.value for option in VoiceOption],
-                value=VoiceOption.Tetiana.value,
+                choices=[voice.title() for voice in voices.keys()],
+                value="Tetiana",
             )
 
     gr.Button("Run").click(