Spaces:
Running
on
Zero
Running
on
Zero
Update InferenceInterfaces/ControllableInterface.py
Browse files
InferenceInterfaces/ControllableInterface.py
CHANGED
@@ -17,6 +17,7 @@ class ControllableInterface:
|
|
17 |
os.environ["CUDA_VISIBLE_DEVICES"] = f"{gpu_id}"
|
18 |
self.device = "cuda" if gpu_id != "cpu" else "cpu"
|
19 |
self.model = ToucanTTSInterface(device=self.device, tts_model_path=None)
|
|
|
20 |
|
21 |
def read(self,
|
22 |
prompt,
|
|
|
17 |
os.environ["CUDA_VISIBLE_DEVICES"] = f"{gpu_id}"
|
18 |
self.device = "cuda" if gpu_id != "cpu" else "cpu"
|
19 |
self.model = ToucanTTSInterface(device=self.device, tts_model_path=None)
|
20 |
+
self.model.set_utterance_embedding("female_high_voice.wav")
|
21 |
|
22 |
def read(self,
|
23 |
prompt,
|