diff --git a/openbb_terminal/forecast/whisper_model.py b/openbb_terminal/forecast/whisper_model.py index aa50e7b5c8a..ebec5c49d18 100644 --- a/openbb_terminal/forecast/whisper_model.py +++ b/openbb_terminal/forecast/whisper_model.py @@ -15,6 +15,7 @@ from transformers import BartForConditionalGeneration, BartTokenizer, pipeline from openbb_terminal.decorators import log_start_end from openbb_terminal.forecast.whisper_utils import slugify, write_srt, write_vtt from openbb_terminal.rich_config import console +from typing import Optional logger = logging.getLogger(__name__) # pylint: disable=too-many-arguments @@ -57,7 +58,7 @@ def transcribe_and_summarize( subtitles_format: str = "vtt", verbose: bool = False, task: str = "transcribe", - language: str = None, + language: Optional[str] = None, breaklines: int = 0, output_dir: str = "", ):