MeBai commited on
Commit
0ddfaed
·
verified ·
1 Parent(s): 370d27b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ processor = WhisperProcessor.from_pretrained(model_name)
9
  model = WhisperForConditionalGeneration.from_pretrained(model_name)
10
 
11
  # 加载数据集 bigcode/the-stack
12
- dataset = load_dataset("CoIR-Retrieval/CodeSearchNet-php-queries-corpus", data_dir="data", split="train")
13
 
14
  def transcribe(audio):
15
  # 处理音频进行转录
 
9
  model = WhisperForConditionalGeneration.from_pretrained(model_name)
10
 
11
  # 加载数据集 bigcode/the-stack
12
+ dataset = load_dataset("CoIR-Retrieval/CodeSearchNet-php-queries-corpus", data_dir="data", split="train[:80%]")
13
 
14
  def transcribe(audio):
15
  # 处理音频进行转录