From 33369a9555316acabf19a07dac2f73c3605534b5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dominik=20Mach=C3=A1=C4=8Dek?= <machacek@ufal.mff.cuni.cz>
Date: Thu, 25 May 2023 14:28:18 +0200
Subject: [PATCH] import WhisperModel

---
 whisper_online.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/whisper_online.py b/whisper_online.py
index 0981911..0f743e0 100644
--- a/whisper_online.py
+++ b/whisper_online.py
@@ -6,6 +6,7 @@ from functools import lru_cache
 import time
 from mosestokenizer import MosesTokenizer
 
+
 @lru_cache
 def load_audio(fname):
     a, _ = librosa.load(fname, sr=16000)
@@ -89,6 +90,8 @@ class FasterWhisperASR(ASRBase):
     sep = ""
 
     def load_model(self, modelsize=None, cache_dir=None, model_dir=None):
+        from faster_whisper import WhisperModel
+
 
         if model_dir is not None:
             print(f"Loading whisper model from model_dir {model_dir}. modelsize and cache_dir parameters are not used.",file=sys.stderr)