stream-translator-gpt 2024.3.3.dev2__tar.gz → 2024.3.3.dev3__tar.gz

{stream-translator-gpt-2024.3.3.dev2/stream_translator_gpt.egg-info → stream-translator-gpt-2024.3.3.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stream-translator-gpt
-Version: 2024.3.3.dev2
+Version: 2024.3.3.dev3
 Summary: Command line tool to transcribe & translate audio from livestreams in real time
 Author-email: ion <ionicbond3@gmail.com>
 Project-URL: Homepage, https://github.com/ionic-bond/stream-translator-gpt

{stream-translator-gpt-2024.3.3.dev2 → stream-translator-gpt-2024.3.3.dev3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "stream-translator-gpt"
-version = "2024.3.3.dev2"
+version = "2024.3.3.dev3"
 authors = [
   { name="ion", email="ionicbond3@gmail.com" },
 ]
@@ -48,6 +48,15 @@ dependencies = [
   "google-generativeai<1.0",
 ]
 
+[project.scripts]
+stream-translator-gpt = "stream_translator_gpt.translator:cli"
+
 [project.urls]
 Homepage = "https://github.com/ionic-bond/stream-translator-gpt"
-Issues = "https://github.com/ionic-bond/stream-translator-gpt/issues"
+Issues = "https://github.com/ionic-bond/stream-translator-gpt/issues"
+
+[tool.setuptools.packages.find]
+where = ["."]
+
+[tool.setuptools.package-data]
+stream_translator_gpt = ["*.jit"]

{stream-translator-gpt-2024.3.3.dev2 → stream-translator-gpt-2024.3.3.dev3}/stream_translator_gpt/audio_slicer.py

@@ -10,13 +10,18 @@ from .common import TranslationTask, SAMPLE_RATE, LoopWorkerBase
 warnings.filterwarnings('ignore')
 
 
+def _init_jit_model(model_path: str, device=torch.device('cpu')):
+    torch.set_grad_enabled(False)
+    model = torch.jit.load(model_path, map_location=device)
+    model.eval()
+    return model
+
+
 class VAD:
 
     def __init__(self):
-        self.model, _ = torch.hub.load(repo_or_dir='snakers4/silero-vad',
-                                       model='silero_vad',
-                                       force_reload=True,
-                                       onnx=False)
+        current_dir = os.path.dirname(__file__)
+        self.model = _init_jit_model(os.path.join(current_dir, 'silero_vad.jit'))
 
     def is_speech(self, audio: np.array, threshold: float = 0.5, sampling_rate: int = 16000):
         if not torch.is_tensor(audio):

{stream-translator-gpt-2024.3.3.dev2 → stream-translator-gpt-2024.3.3.dev3/stream_translator_gpt.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: stream-translator-gpt
-Version: 2024.3.3.dev2
+Version: 2024.3.3.dev3
 Summary: Command line tool to transcribe & translate audio from livestreams in real time
 Author-email: ion <ionicbond3@gmail.com>
 Project-URL: Homepage, https://github.com/ionic-bond/stream-translator-gpt

{stream-translator-gpt-2024.3.3.dev2 → stream-translator-gpt-2024.3.3.dev3}/stream_translator_gpt.egg-info/SOURCES.txt

@@ -11,9 +11,11 @@ stream_translator_gpt/filters.py
 stream_translator_gpt/llm_translator.py
 stream_translator_gpt/print_all_devices.py
 stream_translator_gpt/result_exporter.py
+stream_translator_gpt/silero_vad.jit
 stream_translator_gpt/translator.py
 stream_translator_gpt.egg-info/PKG-INFO
 stream_translator_gpt.egg-info/SOURCES.txt
 stream_translator_gpt.egg-info/dependency_links.txt
+stream_translator_gpt.egg-info/entry_points.txt
 stream_translator_gpt.egg-info/requires.txt
 stream_translator_gpt.egg-info/top_level.txt

stream-translator-gpt-2024.3.3.dev3/stream_translator_gpt.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+stream-translator-gpt = stream_translator_gpt.translator:cli

{stream-translator-gpt-2024.3.3.dev2 → stream-translator-gpt-2024.3.3.dev3}/stream_translator_gpt.egg-info/top_level.txt

@@ -1 +1,2 @@
+dist
 stream_translator_gpt