当前位置：首页 > article >正文

AIGC: 关于ChatGPT中基于Whisper模型实现音频转文本

article 2024/11/15 11:16:11

概述

到目前，GPT只能去接收文本的输入，但是在现实的生活当中，会有语音的需求
GPT也有相关的能力接入，我们就需要一个能够将语音内容转换成文本的能力
- 当然其他第三方的软件或者接口也是支持这个功能
- 在 Open AI 有一个语音转文本的模型叫做 whisper
在 Open AI 它的官方网站当中，在左侧可以看到有一个 Audio 关于音频的API
- 文档: https://platform.openai.com/docs/api-reference/audio/create-transcription
- 接口: https://api.openai.com/v1/audio/transcriptions
可以看到它的参数
- file 参数，必选，file类型，支持 mp3,mp4,mpeg,mpga,m4a,wav,or webm.
- model 参数，必选，string 类型，目前仅支持 whisper-1
- prompt 参数，可选，string 类型
- response_format 参数，可选，string 类型
- temperature 参数，可选，number 类型默认 0
- language 参数，可选，string 类型
- …
与其他的音频转文本的方法或者工具不同的是 Open AI 允许我们去输入 Prompt
- 这个 Prompt 就是对于我们音频的一个介绍
- 举个例子，我们的音频是计算机相关的论文，或者法学，医学相关的论文
- 在这些论文当中，可能会有很多的特有的专业的名词
- 如果我们按照正常行的音频翻译的话，有可能得到的并不是我们想要的结果
- 但是 Open AI 不一样，我们在进行 whisper 调用的时候，可以去向它发送 Prompt
- 这里 Prompt 就是对音频的介绍，我们可以清晰的告诉 GPT，我们的音频是哪个领域，哪个方向
- 然后研究是的是什么的内容等，这样可辅助GPT对音频的内容进行分析转换成较准确的相关的文本，提升转换的质量
response_format 是对于输出进行一个格式化，默认是 Json
这是和音频相关的接口

代码

whisper_example.py

# -*- coding: utf-8
import os
import openai

openai.api_key = "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"
openai.proxy="http://127.0.0.1:7890"

audio_file = open("test.mp3", "rb")
transcript = openai.Audio.transcribe("whisper-1", audio_file, prompt=r"这是一篇机器学习相关的公开课")

print(transcript['text'])

分析

比如，test.mp3 内容是一段机器学习的英文教学章节
整体的代码看上去非常的简单，只需要两步就可以了
接下来执行一下我们刚刚实现的python代码
可以看到音频已经转换成了文本，但是是英文，有时候我们希望它是中文
这里有几个方案
- 方案一，通过其他的翻译软件去进行翻译
- 方案二，可以调用GPT的接口, 通过GBT3.5的模型, 对文章去做一个翻译
  - 把这篇文章作为 Prompt 传递给它，由GPT去做翻译
- 方案三，利用 whisper 去做一个简单的处理
  - 它的输入参数中，有一个 language的输入参数，这里指音频文件输入的语言
  - 这里并不代表我们设置之后会以对应的形式对我们进行输出
  - 我们想要去输出中文怎么办呢？其实，有一个非常简便的方法
  - 在调用 whisper 模型的时候，可以去传入，Prompt, 告诉它这是一篇机器学习相关的公开课
  - 我们通过中文的描述去构建了一个Prompt告诉whisper
  - 由于我们的 Prompt 是输入的中文的描述, 最终whisper给我们返回的结果也是中文的
以上是通过GPT的 whisper 模型将音频转换成文本的一个简单的示例代码