zyu commited on
Commit
3eacb4c
·
1 Parent(s): 03fbdf1
Files changed (1) hide show
  1. app.py +2 -3
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
- print(DATASETS_MODEL_INFO_PATH)
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