AshDavid12 commited on
Commit
aba4a2e
·
1 Parent(s): 9bff659

added flush file

Browse files
Files changed (2) hide show
  1. .gitignore +2 -1
  2. infer.py +2 -0
.gitignore CHANGED
@@ -1 +1,2 @@
1
- *.wav
 
 
1
+ *.wav
2
+ *.ogg
infer.py CHANGED
@@ -270,6 +270,7 @@ async def websocket_transcribe(websocket: WebSocket):
270
  logging.info("Buffered enough audio time, starting transcription.")
271
 
272
  # Create a temporary WAV file in /tmp for transcription
 
273
  with tempfile.NamedTemporaryFile(suffix=".wav", delete=False, dir="/tmp") as temp_wav_file:
274
  logging.info(f"Temporary audio file created at {temp_wav_file.name}")
275
 
@@ -278,6 +279,7 @@ async def websocket_transcribe(websocket: WebSocket):
278
  wav_file.setsampwidth(sample_width)
279
  wav_file.setframerate(sample_rate)
280
  wav_file.writeframes(pcm_audio_buffer)
 
281
 
282
  if not validate_wav_file(temp_wav_file.name):
283
  logging.error(f"Invalid WAV file created: {temp_wav_file.name}")
 
270
  logging.info("Buffered enough audio time, starting transcription.")
271
 
272
  # Create a temporary WAV file in /tmp for transcription
273
+
274
  with tempfile.NamedTemporaryFile(suffix=".wav", delete=False, dir="/tmp") as temp_wav_file:
275
  logging.info(f"Temporary audio file created at {temp_wav_file.name}")
276
 
 
279
  wav_file.setsampwidth(sample_width)
280
  wav_file.setframerate(sample_rate)
281
  wav_file.writeframes(pcm_audio_buffer)
282
+ temp_wav_file.flush()
283
 
284
  if not validate_wav_file(temp_wav_file.name):
285
  logging.error(f"Invalid WAV file created: {temp_wav_file.name}")