Spaces:

lyhourt
/

whisper-small-hu-demo

Sleeping

lyhourt commited on May 24

Commit

d5d2a64

•

1 Parent(s): 639b2fa

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 import torch
 from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
-model_id = "lyhourt/whisper-small-clean_6-v3"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32

 import torch
 from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
+model_id = "lyhourt/whisper-small-clean_6-v4"
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32