whisper-small-ar-demo

Runtime error

MMars commited on Dec 19, 2022

Commit

99e3ce0

•

1 Parent(s): aa6570c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,8 +5,8 @@ import pytube as pt
 from transformers import pipeline
 from huggingface_hub import model_info
-MODEL_NAME = "openai/whisper-small" #this always needs to stay in line 8 :D sorry for the hackiness
-lang = "en"
 device = 0 if torch.cuda.is_available() else "cpu"
 pipe = pipeline(

 from transformers import pipeline
 from huggingface_hub import model_info
+MODEL_NAME = "MMars/whisper-small-ar" #this always needs to stay in line 8 :D sorry for the hackiness
+lang = "ar"
 device = 0 if torch.cuda.is_available() else "cpu"
 pipe = pipeline(