silasyl commited on
Commit
5e739b3
·
1 Parent(s): d2d6430

changed transcription library

Browse files
Files changed (1) hide show
  1. tools.py +10 -2
tools.py CHANGED
@@ -1,6 +1,7 @@
1
  import base64
2
  import cv2
3
  import io
 
4
  import os
5
  import requests
6
  import whisper
@@ -312,9 +313,16 @@ class AudioFileTranscriptTool(Tool):
312
  f.write(response.content)
313
 
314
  # Transcribe the audio
315
- result = self.whisper_model.transcribe("audio.mp3", language="en", fp16=False)
316
 
317
- return result['text']
 
 
 
 
 
 
 
318
 
319
 
320
  class PythonFileDownloader(Tool):
 
1
  import base64
2
  import cv2
3
  import io
4
+ import openai
5
  import os
6
  import requests
7
  import whisper
 
313
  f.write(response.content)
314
 
315
  # Transcribe the audio
316
+ client = openai.OpenAI(api_key=OPENAI_API_KEY)
317
 
318
+ with open("audio.mp3", "rb") as f:
319
+ transcript = client.audio.transcriptions.create(
320
+ model="gpt-4o-mini-transcribe",
321
+ file=f,
322
+ language="en"
323
+ )
324
+
325
+ return transcript.text
326
 
327
 
328
  class PythonFileDownloader(Tool):