From 395b6243ba2bc6aa962c9eb9ceb7385921d9bb83 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 29 Jan 2024 20:42:41 +0100 Subject: [PATCH] updated --- lollms/media.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/lollms/media.py b/lollms/media.py index 89b932d..7dda7b9 100644 --- a/lollms/media.py +++ b/lollms/media.py @@ -90,7 +90,6 @@ class AudioRecorder: self.is_recording = False self.start_time = time.time() self.last_time = time.time() - self.whisper_model = None def audio_callback(self, indata, frames, time_, status): volume_norm = np.linalg.norm(indata)*10 @@ -105,9 +104,6 @@ class AudioRecorder: self.update_spectrogram() def start_recording(self): - if self.whisper_model is None: - self.lollmsCom.info("Loading whisper model") - self.whisper_model=whisper.load_model("base.en") try: self.is_recording = True self.buffer = np.array([], dtype=np.float32) @@ -116,6 +112,7 @@ class AudioRecorder: except Exception as ex: self.lollmsCom.InfoMessage("Couldn't start recording.\nMake sure your input device is connected and operational") trace_exception(ex) + def stop_recording(self): self.is_recording = False self.audio_stream.stop() @@ -123,10 +120,6 @@ class AudioRecorder: write(self.filename, self.sample_rate, self.buffer) self.lollmsCom.info(f"Saved to {self.filename}") self.lollmsCom.info(f"Transcribing ... ") - result = self.whisper_model.transcribe(str(self.filename)) - with open(self.filename.replace("wav","txt"), "w") as f: - f.write(result["text"]) - self.lollmsCom.info(f"Saved to {self.filename}") def update_spectrogram(self): f, t, Sxx = spectrogram(self.buffer[-30*self.sample_rate:], self.sample_rate)