whisper-small-dv-demo

Sleeping

nikbor87 commited on Mar 4

Commit

a2c34e6

•

1 Parent(s): 00a87e1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import pipeline
 from transformers.pipelines.audio_utils import ffmpeg_read
 import gradio as gr
-MODEL_NAME = "openai/whisper-small"
 BATCH_SIZE = 8
 device = 0 if torch.cuda.is_available() else "cpu"

 from transformers.pipelines.audio_utils import ffmpeg_read
 import gradio as gr
+MODEL_NAME = "nikbor87/whisper-small-dv"
 BATCH_SIZE = 8
 device = 0 if torch.cuda.is_available() else "cpu"