Claw commited on
Commit ·
45d96ce
1
Parent(s): 992765b
Fix: use 'prompt' not 'initial_prompt' for Whisper
Browse files
app.py
CHANGED
|
@@ -70,7 +70,7 @@ def transcribe(wav):
|
|
| 70 |
language="zh",
|
| 71 |
task="transcribe",
|
| 72 |
max_new_tokens=256,
|
| 73 |
-
|
| 74 |
)
|
| 75 |
text = processor.batch_decode(ids, skip_special_tokens=True)[0].strip()
|
| 76 |
if text:
|
|
|
|
| 70 |
language="zh",
|
| 71 |
task="transcribe",
|
| 72 |
max_new_tokens=256,
|
| 73 |
+
prompt="无人机 架 大疆 北约 大毛 小前线 杨栋梁"
|
| 74 |
)
|
| 75 |
text = processor.batch_decode(ids, skip_special_tokens=True)[0].strip()
|
| 76 |
if text:
|