bobmaster/matrix-stt-bot
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases 3 Packages Wiki Activity

v1.1.0

Browse source 
Update dependencies

remove useless
This commit is contained in:
hibobmaster 2023-07-20 15:56:01 +08:00
parent d474d14879
commit f20b1d5d40
Signed by: bobmaster
SSH key fingerprint: SHA256:5ZYgd8fg+PcNZNy4SzcSKu5JtqZyBF8kUhY7/k2viDk
2 changed files with 29 additions and 47 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
bot.py
View file

@ -109,30 +109,8 @@ class Bot:
self.to_device_callback, (KeyVerificationEvent, )) self.to_device_callback, (KeyVerificationEvent, ))
# intialize whisper model # intialize whisper model
if not os.path.exists(os.path.join(self.download_root, "model.txt")):
# that means we have not downloaded the model yet
logger.info("Model downloading......")
self.model_size_or_path = self.model_size
with open(os.path.join(self.download_root, "model.txt"), "w") as f:
f.write(self.model_size)
f.close()
else:
# model exists
f = open(os.path.join(self.download_root, "model.txt"), "r")
old_model = f.read()
if old_model != self.model_size:
# model size changed
self.model_size_or_path = self.model_size
f.close()
with open(os.path.join(self.download_root, "model.txt"), "w") as f:
f.write(self.model_size)
f.close()
else:
# use existed model
self.model_size_or_path = self.download_root
self.model = WhisperModel( self.model = WhisperModel(
model_size_or_path=self.model_size_or_path, model_size_or_path=self.model_size,
device=self.device, device=self.device,
compute_type=self.compute_type, compute_type=self.compute_type,
cpu_threads=self.cpu_threads, cpu_threads=self.cpu_threads,

52
requirements.txt
View file

@ -1,5 +1,5 @@
aiofiles==23.1.0 aiofiles==23.1.0
aiohttp==3.8.4 aiohttp==3.8.5
aiohttp-socks==0.7.1 aiohttp-socks==0.7.1
aiosignal==1.3.1 aiosignal==1.3.1
async-timeout==4.0.2 async-timeout==4.0.2
@ -7,44 +7,48 @@ atomicwrites==1.4.1
attrs==23.1.0 attrs==23.1.0
av==10.0.0 av==10.0.0
cachetools==4.2.4 cachetools==4.2.4
certifi==2022.12.7 certifi==2023.5.7
cffi==1.15.1 cffi==1.15.1
charset-normalizer==3.1.0 charset-normalizer==3.2.0
coloredlogs==15.0.1 coloredlogs==15.0.1
ctranslate2==3.12.0 ctranslate2==3.17.0
faster-whisper @ git+https://github.com/guillaumekln/faster-whisper.git@3adcc12d0f91369446a624e33185c555facc8ed2 faster-whisper==0.7.0
filelock==3.12.0 filelock==3.12.2
flatbuffers==23.3.3 flatbuffers==23.5.26
frozenlist==1.3.3 frozenlist==1.4.0
fsspec==2023.6.0
future==0.18.3 future==0.18.3
h11==0.14.0 h11==0.14.0
h2==4.1.0 h2==4.1.0
hpack==4.0.0 hpack==4.0.0
huggingface-hub @ git+https://github.com/huggingface/huggingface_hub.git@a433410ac37f0efaacb92b09684efa3ae46ed3fd huggingface-hub==0.16.4
humanfriendly==10.0 humanfriendly==10.0
hyperframe==6.0.1 hyperframe==6.0.1
idna==3.4 idna==3.4
jsonschema==4.17.3 jsonschema==4.18.4
jsonschema-specifications==2023.7.1
Logbook==1.5.3 Logbook==1.5.3
matrix-nio[e2e]==0.20.2 matrix-nio==0.21.2
mpmath==1.3.0 mpmath==1.3.0
multidict==6.0.4 multidict==6.0.4
numpy==1.24.2 numpy==1.25.1
onnxruntime==1.14.1 onnxruntime==1.15.1
packaging==23.1 packaging==23.1
peewee==3.16.1 peewee==3.16.2
protobuf==4.22.3 protobuf==4.23.4
pycparser==2.21 pycparser==2.21
pycryptodome==3.17 pycryptodome==3.18.0
pyrsistent==0.19.3 pyrsistent==0.19.3
python-olm==3.1.3 python-olm==3.2.15
python-socks==2.2.0 python-socks==2.3.0
PyYAML==6.0 PyYAML==6.0.1
requests==2.28.2 referencing==0.30.0
sympy==1.11.1 requests==2.31.0
rpds-py==0.9.2
sympy==1.12
tokenizers==0.13.3 tokenizers==0.13.3
tqdm==4.65.0 tqdm==4.65.0
typing_extensions==4.5.0 typing_extensions==4.7.1
unpaddedbase64==2.1.0 unpaddedbase64==2.1.0
urllib3==1.26.15 urllib3==2.0.4
yarl==1.8.2 yarl==1.9.2