sugarcase
2024-09-25 8cd8e25f180b57c74f1ae7fd39001e5af4c20f8b
funasr/models/whisper/model.py
@@ -7,7 +7,11 @@
import torch.nn.functional as F
from torch import Tensor
from torch import nn
import whisper
# import whisper_timestamped as whisper
from funasr.utils.load_utils import load_audio_text_image_video, extract_fbank
from funasr.register import tables
@@ -108,7 +112,9 @@
        # decode the audio
        options = whisper.DecodingOptions(**kwargs.get("DecodingOptions", {}))
        result = whisper.decode(self.model, speech, options)
        result = whisper.decode(self.model, speech, options=options)
        # result = whisper.transcribe(self.model, speech)
        results = []
        result_i = {"key": key[0], "text": result.text}