ExternalVendorCode/ecoute
1
0
Fork 0
mirror of https://github.com/SevaSk/ecoute.git synced 2024-12-24 06:46:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

return nothing on error in get_transcription

Browse Source
This commit is contained in:
SevaSk 2023-05-30 07:37:56 -04:00
parent 0227cc7fcb
commit acff32780b
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
TranscriberModels.py
View File

@ -19,6 +19,7 @@ class WhisperTranscriber:
result = self.audio_model.transcribe(wav_file_path, fp16=torch.cuda.is_available()) result = self.audio_model.transcribe(wav_file_path, fp16=torch.cuda.is_available())
except Exception as e: except Exception as e:
print(e) print(e)
return ''
return result['text'].strip() return result['text'].strip()
class APIWhisperTranscriber: class APIWhisperTranscriber:
@ -30,5 +31,6 @@ class APIWhisperTranscriber:
result = openai.Audio.translate("whisper-1", audio_file) result = openai.Audio.translate("whisper-1", audio_file)
except Exception as e: except Exception as e:
print(e) print(e)
return ''
return result['text'].strip() return result['text'].strip()