Hotifixing fairseq. (#581)

Narsil · web-flow · commit 46843f5bb34b · 2022-01-07T18:12:38.000+01:00
diff --git a/api-inference-community/docker_images/fairseq/app/pipelines/audio_to_audio.py b/api-inference-community/docker_images/fairseq/app/pipelines/audio_to_audio.py
@@ -75,4 +75,4 @@ def __call__(self, inputs: np.array) -> Tuple[np.array, int, List[str]]:
             wav, sr = TTSHubInterface.get_prediction(
                 self.tts_task, self.tts_model, self.tts_generator, tts_sample
             )
-            return wav.numpy(), sr, [text]
+            return wav.unsqueeze(0).numpy(), sr, [text]

Original file line number	Diff line number	Diff line change
`@@ -75,4 +75,4 @@ def __call__(self, inputs: np.array) -> Tuple[np.array, int, List[str]]:`
`75`	`75`	`wav, sr = TTSHubInterface.get_prediction(`
`76`	`76`	`self.tts_task, self.tts_model, self.tts_generator, tts_sample`
`77`	`77`	`)`
`78`		`- return wav.numpy(), sr, [text]`
	`78`	`+ return wav.unsqueeze(0).numpy(), sr, [text]`