Update functions.py
Browse files- functions.py +1 -1
functions.py
CHANGED
@@ -30,7 +30,7 @@ margin-bottom: 2.5rem">{}</div> """
|
|
30 |
@st.experimental_singleton(suppress_st_warning=True)
|
31 |
def load_models():
|
32 |
asr_model = whisper.load_model("small")
|
33 |
-
asr_pipe = pipeline("
|
34 |
q_model = ORTModelForSequenceClassification.from_pretrained("nickmuchi/quantized-optimum-finbert-tone")
|
35 |
ner_model = AutoModelForTokenClassification.from_pretrained("xlm-roberta-large-finetuned-conll03-english")
|
36 |
q_tokenizer = AutoTokenizer.from_pretrained("nickmuchi/quantized-optimum-finbert-tone")
|
|
|
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-large")
|
34 |
q_model = ORTModelForSequenceClassification.from_pretrained("nickmuchi/quantized-optimum-finbert-tone")
|
35 |
ner_model = AutoModelForTokenClassification.from_pretrained("xlm-roberta-large-finetuned-conll03-english")
|
36 |
q_tokenizer = AutoTokenizer.from_pretrained("nickmuchi/quantized-optimum-finbert-tone")
|