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

fix invalid cross-device link

Browse source
This commit is contained in:
hibobmaster 2023-04-20 01:51:00 +08:00
parent 70d8a8aa0c
commit 953d521a37
Signed by: bobmaster
GPG key ID: 316B77D7914D713C
2 changed files with 11 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
bot.py
View file

@ -19,8 +19,6 @@ from faster_whisper import WhisperModel
from log import getlogger from log import getlogger
from send_message import send_room_message from send_message import send_room_message
from unsync import unsync
logger = getlogger() logger = getlogger()
@ -82,9 +80,10 @@ class Bot:
self.num_workers = 1 self.num_workers = 1
if download_root is None: if download_root is None:
self.download_root = "models" cwd = os.getcwd()
if not os.path.exists("models"): self.download_root = os.path.join(cwd, "models")
os.mkdir("models") if not os.path.exists(self.download_root):
os.mkdir(self.download_root)
# initialize AsyncClient object # initialize AsyncClient object
self.store_path = os.getcwd() self.store_path = os.getcwd()
@ -112,10 +111,11 @@ class Bot:
# intialize whisper model # intialize whisper model
if not os.path.exists(os.path.join(self.download_root, "model.txt")): if not os.path.exists(os.path.join(self.download_root, "model.txt")):
# that means we have not downloaded the model yet # that means we have not downloaded the model yet
logger.info("Model downloading") logger.info("Model downloading......")
self.model_size_or_path = self.model_size self.model_size_or_path = self.model_size
with open(os.path.join(self.download_root, "model.txt"), "w") as f: with open(os.path.join(self.download_root, "model.txt"), "w") as f:
f.write(self.model_size) f.write(self.model_size)
f.close()
else: else:
# model exists # model exists
f = open(os.path.join(self.download_root, "model.txt"), "r") f = open(os.path.join(self.download_root, "model.txt"), "r")
@ -508,7 +508,8 @@ class Bot:
if isinstance(resp, EncryptionError): if isinstance(resp, EncryptionError):
logger.error(f"import_keys failed with {resp}") logger.error(f"import_keys failed with {resp}")
else: else:
logger.info(f"import_keys success, please remove import_keys configuration!!!") logger.info(
f"import_keys success, please remove import_keys configuration!!!")
# whisper function # whisper function
def transcribe(self, filename: str) -> str: def transcribe(self, filename: str) -> str:

3
requirements.txt
View file

@ -20,7 +20,7 @@ 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==0.13.4 huggingface-hub @ git+https://github.com/huggingface/huggingface_hub.git@a433410ac37f0efaacb92b09684efa3ae46ed3fd
humanfriendly==10.0 humanfriendly==10.0
hyperframe==6.0.1 hyperframe==6.0.1
idna==3.4 idna==3.4
@ -46,6 +46,5 @@ tokenizers==0.13.3
tqdm==4.65.0 tqdm==4.65.0
typing_extensions==4.5.0 typing_extensions==4.5.0
unpaddedbase64==2.1.0 unpaddedbase64==2.1.0
unsync==1.4.0
urllib3==1.26.15 urllib3==1.26.15
yarl==1.8.2 yarl==1.8.2