owaski commited on
Commit
cc5e2cb
·
1 Parent(s): 4d5c730

whisper median and 12l mRASP2

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import gradio as gr
6
  import whisper
7
 
8
  # the model we are using for ASR, options are small, medium, large and largev2 (large and largev2 don't fit on huggingface cpu)
9
- model = whisper.load_model("small")
10
 
11
  import torch
12
 
@@ -28,7 +28,7 @@ os.system("pip install -r requirements.txt")
28
  os.system("git clone https://github.com/pytorch/fairseq")
29
  os.system("cd fairseq; pip install ./; cd ..")
30
 
31
- model_name = "6e6d_no_mono.pt"
32
  # os.system("wget https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/acl2021/mrasp2/" + model_name)
33
  os.system("wget https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/acl2021/mrasp2/bpe_vocab")
34
  os.system("wget https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/emnlp2020/mrasp/pretrain/dataset/codes.bpe.32000")
 
6
  import whisper
7
 
8
  # the model we are using for ASR, options are small, medium, large and largev2 (large and largev2 don't fit on huggingface cpu)
9
+ model = whisper.load_model("medium")
10
 
11
  import torch
12
 
 
28
  os.system("git clone https://github.com/pytorch/fairseq")
29
  os.system("cd fairseq; pip install ./; cd ..")
30
 
31
+ model_name = "12e12d_no_mono.pt"
32
  # os.system("wget https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/acl2021/mrasp2/" + model_name)
33
  os.system("wget https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/acl2021/mrasp2/bpe_vocab")
34
  os.system("wget https://lf3-nlp-opensource.bytetos.com/obj/nlp-opensource/emnlp2020/mrasp/pretrain/dataset/codes.bpe.32000")