MeBai commited on
Commit
7c5454f
·
verified ·
1 Parent(s): f52c390

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,13 +4,13 @@ from datasets import load_dataset
4
  import torch
5
 
6
  # 加载 Whisper 模型和 processor
7
- model_name = "openai/whisper-small"
8
  processor = WhisperProcessor.from_pretrained(model_name)
9
  model = WhisperForConditionalGeneration.from_pretrained(model_name)
10
 
11
  # 加载数据集 bigcode/the-stack
12
 
13
- #ds = load_dataset("CoIR-Retrieval/CodeSearchNet-php-queries-corpus")
14
 
15
  def transcribe(audio):
16
  # 处理音频进行转录
 
4
  import torch
5
 
6
  # 加载 Whisper 模型和 processor
7
+ model_name = "openai/whisper-large-v3-turbo"
8
  processor = WhisperProcessor.from_pretrained(model_name)
9
  model = WhisperForConditionalGeneration.from_pretrained(model_name)
10
 
11
  # 加载数据集 bigcode/the-stack
12
 
13
+ ds = load_dataset("CoIR-Retrieval/CodeSearchNet-php-queries-corpus")
14
 
15
  def transcribe(audio):
16
  # 处理音频进行转录