Update app.py

app.py

@@ -50,7 +50,7 @@ try:
   
   # Prepare the dataset
   dataset = dataset.map(prepare_dataset, remove_columns=dataset.column_names)
-  example = dataset[10]
+  example = dataset[5]
   speaker_embeddings = torch.tensor(example["speaker_embeddings"]).unsqueeze(0)
   
 except Exception as e:
@@ -149,11 +149,10 @@ def text_preprocessing(text):
     text = ' '.join(text.split())
     for src, dst in replacements:
         text = text.replace(src, dst)
-    return text
     
-inputs = processor(text=text, return_tensors="pt")
-speech = model.generate_speech(inputs["input_ids"], speaker_embedding, vocoder=vocoder)
-return (16000, speech.numpy())
+    inputs = processor(text=text, return_tensors="pt")
+    speech = model.generate_speech(inputs["input_ids"], speaker_embedding, vocoder=vocoder)
+    return (16000, speech.numpy())
 
 iface = gr.Interface(
     fn=text_to_speech,