From 037ef8bc41a9c3dd0d09921fdea815ce8f3e9b3c Mon Sep 17 00:00:00 2001
From: Adam Siemiginowski <adamsiem@users.noreply.github.com>
Date: Wed, 21 Jun 2023 13:29:37 -0400
Subject: [PATCH] set default whisper model to tiny + import nltk

---
 config.ini          | 2 +-
 whisjax_realtime.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/config.ini b/config.ini
index 138a1778..62699b42 100644
--- a/config.ini
+++ b/config.ini
@@ -5,7 +5,7 @@ KMP_DUPLICATE_LIB_OK=TRUE
 OPENAI_APIKEY=
 # Export Whisper Model Size
 WHISPER_MODEL_SIZE=medium
-WHISPER_REAL_TIME_MODEL_SIZE=medium
+WHISPER_REAL_TIME_MODEL_SIZE=tiny
 # AWS config
 AWS_ACCESS_KEY=***REMOVED***
 AWS_SECRET_KEY=***REMOVED***
diff --git a/whisjax_realtime.py b/whisjax_realtime.py
index 7a93cee5..827368fa 100644
--- a/whisjax_realtime.py
+++ b/whisjax_realtime.py
@@ -11,7 +11,8 @@ from file_utilities import upload_files
 from viz_utilities import create_wordcloud, create_talk_diff_scatter_viz
 from text_utilities import summarize, post_process_transcription
 from loguru import logger
-
+import nltk
+nltk.download('stopwords')
 
 config = configparser.ConfigParser()
 config.read('config.ini')