ExpiredPopsicle/whisper-live
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed 'FP16 is not supported on CPU' Warning

Browse Source
This commit is contained in:
John Ciubuc 2023-01-14 22:13:48 -06:00
parent 4c0d6821ff
commit f180571916
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
transcribe_demo.py
View File

@ -5,6 +5,7 @@ import io
import os import os
import speech_recognition as sr import speech_recognition as sr
import whisper import whisper
import torch
from datetime import datetime, timedelta from datetime import datetime, timedelta
from queue import Queue from queue import Queue
@ -99,7 +100,7 @@ def main():
f.write(wav_data.read()) f.write(wav_data.read())
# Read the transcription. # Read the transcription.
result = audio_model.transcribe(temp_file) result = audio_model.transcribe(temp_file, fp16=torch.cuda.is_available())
text = result['text'].strip() text = result['text'].strip()
# If we detected a pause between recordings, add a new item to our transcripion. # If we detected a pause between recordings, add a new item to our transcripion.