Merge branch 'main' of github.com:abdeladim-s/subsai

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "subsai"
-version = "1.2.1"
+version = "1.2.2"
 authors = [
     {name = "abdeladim-s"},
 ]

requirements.txt

@@ -4,10 +4,10 @@ streamlit~=1.20.0
 streamlit_player~=0.1.5
 streamlit-aggrid~=0.3.4
 ffsubsync~=0.4.24
-whisper-timestamped @ git+https://github.com/linto-ai/whisper-timestamped@127e8b738756420487210490d09938593244b3dd
+whisper-timestamped @ git+https://github.com/linto-ai/whisper-timestamped
 pandas==1.5.3
 pysubs2~=1.6.0
 pywhispercpp==1.1.1
 dl_translate==0.2.6
 faster_whisper==0.5.1
-whisperx @ git+https://github.com/m-bain/whisperx@0ae0d49d1dd789322181f579c71719f4cac83ae1
+whisperx @ git+https://github.com/m-bain/whisperx@0ae0d49d1dd789322181f579c71719f4cac83ae1

src/subsai/models/whisperX_model.py

@@ -33,9 +33,9 @@ class WhisperXModel(AbstractModel):
         },
         'device': {
             'type': list,
-            'description': "The PyTorch device to put the model into",
-            'options': [None, *get_available_devices()],
-            'default': None
+            'description': 'Device to use for computation ("cpu", "cuda")',
+            'options': ['cpu', 'cuda'],
+            'default': 'cpu'
         },
         'compute_type': {
             'type': list,

src/subsai/webui.py

@@ -515,7 +515,7 @@ def webui() -> None:
             exported_file = media_file.parent / (export_filename + export_format)
             subs.save(exported_file, fps=fps)
             st.success(f'Exported file to {exported_file}', icon="✅")
-            with open(exported_file, 'r') as f:
+            with open(exported_file, 'r', encoding='utf-8') as f:
                 st.download_button('Download', f, file_name=export_filename + export_format)
 
     with st.expander('Merge subtitles with video'):