Spaces:

Ritesh-hf
/

speech-to-text-with-timestamps

Sleeping

Ritesh-hf commited on Jun 17, 2024

Commit

b7b9c25

verified ·

1 Parent(s): b6d60a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import io
 import torch
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
-model_id = "openai/whisper-tiny"
 model = AutoModelForSpeechSeq2Seq.from_pretrained(
     model_id, low_cpu_mem_usage=True, use_safetensors=True

 import torch
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
+model_id = "openai/whisper-large-v3"
 model = AutoModelForSpeechSeq2Seq.from_pretrained(
     model_id, low_cpu_mem_usage=True, use_safetensors=True