romanian-whisper

Runtime error

gigant commited on Dec 12, 2022

Commit

4e08008

•

1 Parent(s): 43e2000

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 = "gigant/whisper-medium-romanian" #this always needs to stay in line 8 :D sorry for the hackiness
+lang = "ro"
 device = 0 if torch.cuda.is_available() else "cpu"
 pipe = pipeline(