whisper-demo-french

Runtime error

pierreguillou commited on Dec 9, 2022

Commit

3bd5a94

1 Parent(s): 01e6207

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"

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