Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ def load_models():
|
|
16 |
st.session_state.translation_tokenizer = AutoTokenizer.from_pretrained("botisan-ai/mt5-translate-yue-zh")
|
17 |
st.session_state.translation_model = AutoModelForSeq2SeqLM.from_pretrained("botisan-ai/mt5-translate-yue-zh")
|
18 |
|
19 |
-
st.session_state.rating_pipe = pipeline("
|
20 |
|
21 |
def transcribe_audio(audio_path):
|
22 |
pipe = st.session_state.transcription_pipe
|
|
|
16 |
st.session_state.translation_tokenizer = AutoTokenizer.from_pretrained("botisan-ai/mt5-translate-yue-zh")
|
17 |
st.session_state.translation_model = AutoModelForSeq2SeqLM.from_pretrained("botisan-ai/mt5-translate-yue-zh")
|
18 |
|
19 |
+
st.session_state.rating_pipe = pipeline("sentiment-analysis", model="uer/roberta-base-finetuned-dianping-chinese")
|
20 |
|
21 |
def transcribe_audio(audio_path):
|
22 |
pipe = st.session_state.transcription_pipe
|