Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 from transformers import SpeechT5Processor, SpeechT5ForTextToSpeech, SpeechT5HifiGan
 
 
-checkpoint = "microsoft/speecht5_tts"
+checkpoint = "GreenCounsel/speecht5_tts_common_voice_5_sv"
 processor = SpeechT5Processor.from_pretrained(checkpoint)
 model = SpeechT5ForTextToSpeech.from_pretrained(checkpoint)
 vocoder = SpeechT5HifiGan.from_pretrained("microsoft/speecht5_hifigan")
@@ -31,23 +31,7 @@ def predict(text, speaker):
     input_ids = inputs["input_ids"]
     input_ids = input_ids[..., :model.config.max_text_positions]
 
-    if speaker == "Surprise Me!":
-        # load one of the provided speaker embeddings at random
-        idx = np.random.randint(len(speaker_embeddings))
-        key = list(speaker_embeddings.keys())[idx]
-        speaker_embedding = np.load(speaker_embeddings[key])
-
-        # randomly shuffle the elements
-        np.random.shuffle(speaker_embedding)
-
-        # randomly flip half the values
-        x = (np.random.rand(512) >= 0.5) * 1.0
-        x[x == 0] = -1.0
-        speaker_embedding *= x
-
-        #speaker_embedding = np.random.rand(512).astype(np.float32) * 0.3 - 0.15
-    else:
-        speaker_embedding = np.load(speaker_embeddings[speaker[:3]])
+    speaker_embedding = np.load(speaker_embeddings[speaker[:3]])
 
     speaker_embedding = torch.tensor(speaker_embedding).unsqueeze(0)
 
@@ -91,12 +75,10 @@ article = """
 """
 
 examples = [
-    ["It is not in the stars to hold our destiny but in ourselves.", "BDL (male)"],
-    ["The octopus and Oliver went to the opera in October.", "CLB (female)"],
-    ["She sells seashells by the seashore. I saw a kitten eating chicken in the kitchen.", "RMS (male)"],
-    ["Brisk brave brigadiers brandished broad bright blades, blunderbusses, and bludgeons—balancing them badly.", "SLT (female)"],
-    ["A synonym for cinnamon is a cinnamon synonym.", "BDL (male)"],
-    ["How much wood would a woodchuck chuck if a woodchuck could chuck wood? He would chuck, he would, as much as he could, and chuck as much wood as a woodchuck would if a woodchuck could chuck wood.", "CLB (female)"],
+    ["GreenCounsel grundades i Malmö för sex år sedan.", "CLB (female)"],
+    ["Med hjälp av maskininlärning kan mycket av juridiken automatiseras samtidigt som juristerna fokuserar på frågor där de ger störst värde.GreenCounsel har byggt en chatbott som kan förstå frågor på många olika språk och ge kvalitetssäkrade svar.She sells seashells by the seashore. I saw a kitten eating chicken in the kitchen.", "CLB (female)"],
+    ["GreenCounsel har byggt en chatbott som kan förstå frågor på många olika språk och ge kvalitetssäkrade svar.", "CLB (female)"],
+    ["Vi har också byggt ett system för att automatisera arbetsflöden för juridiska tjänster via internet.", "CLB (female)"],
 ]
 
 gr.Interface(
@@ -104,14 +86,10 @@ gr.Interface(
     inputs=[
         gr.Text(label="Input Text"),
         gr.Radio(label="Speaker", choices=[
-            "BDL (male)",
             "CLB (female)",
-            "KSP (male)",
-            "RMS (male)",
-            "SLT (female)",
-            "Surprise Me!"
+          
         ],
-        value="BDL (male)"),
+        value="CLB (female)"),
     ],
     outputs=[
         gr.Audio(label="Generated Speech", type="numpy"),