Method predict

predict.py:20–31

Run a single prediction on the model

(
        self,
        audio_file: Path = Input(description="Input audio file."),
        text: str = Input(description="Input text.", default="water drops"),
    )

        )

    def predict(
        self,
        audio_file: Path = Input(description="Input audio file."),
        text: str = Input(description="Input text.", default="water drops"),
    ) -> Path:
        """Run a single prediction on the model"""

        output_file = "/tmp/separated_audio.wav"

        # AudioSep processes the audio at 32 kHz sampling rate
        inference(self.model, str(audio_file), text, output_file, "cuda")
        return Path(output_file)