cotxetj commited on
Commit
4c84112
1 Parent(s): c6fcce4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -7
app.py CHANGED
@@ -27,14 +27,12 @@ def inference(audio):
27
 
28
 
29
  # Load Whisper-small
30
- pipe = pipeline("automatic-speech-recognition",
31
- model="openai/whisper-small",
32
- device=device
33
- )
 
34
 
35
- # Load the model checkpoint and tokenizer
36
- #model = VitsModel.from_pretrained("Matthijs/mms-tts-eng")
37
- #tokenizer = VitsTokenizer.from_pretrained("Matthijs/mms-tts-eng")
38
  model2 = VitsModel.from_pretrained("facebook/mms-tts-eng")
39
  tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-eng")
40
 
 
27
 
28
 
29
  # Load Whisper-small
30
+ # pipe = pipeline("automatic-speech-recognition",
31
+ # model="openai/whisper-small",
32
+ # device=device
33
+ # )
34
+ pipe = pipeline(model="Sleepyp00/whisper-small-Swedish")
35
 
 
 
 
36
  model2 = VitsModel.from_pretrained("facebook/mms-tts-eng")
37
  tokenizer = VitsTokenizer.from_pretrained("facebook/mms-tts-eng")
38