Spaces:
Sleeping
Sleeping
zyu
commited on
Commit
·
3eacb4c
1
Parent(s):
03fbdf1
style:
Browse files
app.py
CHANGED
@@ -163,7 +163,7 @@ def main():
|
|
163 |
|
164 |
msg_model = "Please confirm model selection via the \'Select Model\' Button first!" \
|
165 |
if st.session_state.model_state['current_config'] is None \
|
166 |
-
else f"Current Model:{st.session_state.model_state['current_config']}"
|
167 |
|
168 |
st.write(msg_model)
|
169 |
|
@@ -190,7 +190,6 @@ def main():
|
|
190 |
st.session_state.translate_in_progress = True
|
191 |
with right:
|
192 |
with st.spinner("Translating..."):
|
193 |
-
input_text = st.session_state.model_state['current_config'] + input_text
|
194 |
prediction = translate(input_text, model, tokenizer, src_lang, tgt_lang)
|
195 |
|
196 |
st.session_state.translation_result['input'] = input_text
|
@@ -202,7 +201,7 @@ def main():
|
|
202 |
|
203 |
if __name__ == '__main__':
|
204 |
DATASETS_MODEL_INFO_PATH = os.path.join(os.getcwd(), "dataset_and_model_info.json")
|
205 |
-
|
206 |
DATASETS_MODEL_INFO = load_json(DATASETS_MODEL_INFO_PATH)
|
207 |
DEFAULT_MODEL = 'google/mt5-small'
|
208 |
|
|
|
163 |
|
164 |
msg_model = "Please confirm model selection via the \'Select Model\' Button first!" \
|
165 |
if st.session_state.model_state['current_config'] is None \
|
166 |
+
else f"Current Model: {st.session_state.model_state['current_config']}"
|
167 |
|
168 |
st.write(msg_model)
|
169 |
|
|
|
190 |
st.session_state.translate_in_progress = True
|
191 |
with right:
|
192 |
with st.spinner("Translating..."):
|
|
|
193 |
prediction = translate(input_text, model, tokenizer, src_lang, tgt_lang)
|
194 |
|
195 |
st.session_state.translation_result['input'] = input_text
|
|
|
201 |
|
202 |
if __name__ == '__main__':
|
203 |
DATASETS_MODEL_INFO_PATH = os.path.join(os.getcwd(), "dataset_and_model_info.json")
|
204 |
+
logger.info(DATASETS_MODEL_INFO_PATH)
|
205 |
DATASETS_MODEL_INFO = load_json(DATASETS_MODEL_INFO_PATH)
|
206 |
DEFAULT_MODEL = 'google/mt5-small'
|
207 |
|