#60 updated requirements.txt, test cases, audio services, added API v…

…ersioning
joshsoftware · Dec 28, 2024 · 448645d · 448645d
1 parent 4808cba
commit 448645d
Show file tree

Hide file tree

Showing 16 changed files with 111 additions and 449 deletions.
diff --git a/app.py b/app.py
diff --git a/config.py b/config.py
diff --git a/download_model.py b/download_model.py
diff --git a/download_whisper.py b/download_whisper.py
diff --git a/extract_entities.py b/extract_entities.py
diff --git a/load_model.py b/load_model.py
diff --git a/logger.py b/logger.py
diff --git a/logging.connf b/logging.connf
diff --git a/requirements.txt b/requirements.txt
@@ -20,3 +20,4 @@ google-api-python-client
 python-dotenv
 whisper-timestamped
 silero-vad
+fastapi_versionizer
diff --git a/service/audio_service.py b/service/audio_service.py
@@ -21,7 +21,7 @@
 #   journal={arXiv preprint arXiv:2212.04356},
 #   year={2022}
 # }
-# @article{JSSv031i07,
+# @article{JSSv031i07,\
 #   title={Computing and Visualizing Dynamic Time Warping Alignments in R: The dtw Package},
 #   author={Giorgino, Toni},
 #   journal={Journal of Statistical Software},

diff --git a/service/load_model.py b/service/load_model.py
@@ -1,12 +1,16 @@
 import torch
 import whisper
+import whisper_timestamped
 
 #load the whisper model from net if it isn't stored locally
 def load_model(model_id, model_path, is_ts):
     #check GPU is avaialbe
     device = "cuda" if torch.cuda.is_available() else "cpu"
     #device = "cpu"
-    model = whisper.load_model(model_id, device=device, download_root=model_path,)
+    if (is_ts):
+        model = whisper_timestamped.load_model(model_id, device=device, download_root=model_path)
+    else:
+        model = whisper.load_model(model_id, device=device, download_root=model_path)
     print(
         f"Model will be run on {device}\n"
         f"Model is {'multilingual' if model.is_multilingual else 'English-only'} "