RasmusToivanen commited on
Commit
a755a71
1 Parent(s): c2ae77e

change to faster model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ from fastT5 import export_and_get_onnx_model
16
  import torch
17
  from transformers import pipeline
18
 
19
- MODEL = "Finnish-NLP/wav2vec2-large-uralic-voxpopuli-v2-finnish"
20
  marian_nmt_model = "Helsinki-NLP/opus-mt-tc-big-fi-en"
21
  tokenizer_marian = MarianTokenizer.from_pretrained(marian_nmt_model)
22
  model = MarianMTModel.from_pretrained(marian_nmt_model)
16
  import torch
17
  from transformers import pipeline
18
 
19
+ MODEL = "Finnish-NLP/wav2vec2-base-fi-voxpopuli-v2-finetuned"
20
  marian_nmt_model = "Helsinki-NLP/opus-mt-tc-big-fi-en"
21
  tokenizer_marian = MarianTokenizer.from_pretrained(marian_nmt_model)
22
  model = MarianMTModel.from_pretrained(marian_nmt_model)