Merge pull request #7 from JuergenFleiss/update_fasterwhisper_pyannote

Update fasterwhisper and pyannote

README.md

@@ -81,17 +81,17 @@ Then you need to clone our repository into that directory including the extra in
 
 **Windows**
 ```
-pip install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git --extra-index-url https://download.pytorch.org/whl/cu118
+pip install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git@update_fasterwhisper_pyannote --extra-index-url https://download.pytorch.org/whl/cu121
 ```
 **Debian** 
 ```
-pip3 install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git --extra-index-url https://download.pytorch.org/whl/cu118
+pip3 install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git@update_fasterwhisper_pyannote --extra-index-url https://download.pytorch.org/whl/cu121
 ```
 💡 Linux keeps killing collection of torch? Try adding the flag--no-cache-dir
 
 **MacOS** 
 ```
-pip3 install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git
+pip3 install aTrain_core@git+https://github.com/JuergenFleiss/aTrain_core.git@update_fasterwhisper_pyannote
 ```
 💡 Note: At the moment, NVIDIA CUDA GPU support is only available for Windows and Debian. We are working on speeding up the transcriptions on MacOS. Hence no need to install the extra index on MacOS. 
 

setup.py

@@ -4,9 +4,9 @@
 
 system = platform.system()
 if system in ["Windows","Linux"]:
-    torch = "torch==2.0.0+cu118"
+    torch = "torch==2.2.0+cu121"
 if system == "Darwin":
-    torch = "torch==2.0.0"
+    torch = "torch==2.2.0"
 
 main_ns = {}
 ver_path = convert_path('aTrain_core/version.py')
@@ -22,17 +22,17 @@
     python_requires=">=3.10",
     install_requires=[
         torch,
-        "torchaudio==2.0.1",
-        "faster-whisper>=0.8",
+        "torchaudio==2.2.0",
+        "faster-whisper==1.0.2",
         "transformers",
         "ffmpeg-python>=0.2",
         "pandas",
-        "pyannote.audio==3.0.0"],
+        "pyannote.audio==3.2.0"],
     packages=find_packages(),
     include_package_data=True,
     entry_points={
         'console_scripts': ['aTrain_core = aTrain_core.cli:cli',]
     }
 )
-
+#