nickmuchi commited on
Commit
8a26ab3
1 Parent(s): b3059bf

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -29,7 +29,7 @@ margin-bottom: 2.5rem">{}</div> """
29
 
30
  @st.experimental_singleton(suppress_st_warning=True)
31
  def load_models():
32
- asr_model = whisper.load_model("base")
33
  #asr_pipe = pipeline("automatic-speech-recognition",model = "openai/whisper-small")
34
  q_model = ORTModelForSequenceClassification.from_pretrained("nickmuchi/quantized-optimum-finbert-tone")
35
  ner_model = AutoModelForTokenClassification.from_pretrained("xlm-roberta-large-finetuned-conll03-english")
 
29
 
30
  @st.experimental_singleton(suppress_st_warning=True)
31
  def load_models():
32
+ asr_model = whisper.load_model("small")
33
  #asr_pipe = pipeline("automatic-speech-recognition",model = "openai/whisper-small")
34
  q_model = ORTModelForSequenceClassification.from_pretrained("nickmuchi/quantized-optimum-finbert-tone")
35
  ner_model = AutoModelForTokenClassification.from_pretrained("xlm-roberta-large-finetuned-conll03-english")