Spaces:

melnguyen
/

en-to-vn-translator

Runtime error

Mel Nguyen (she/her) commited on Oct 22, 2022

Commit

e8da2f4

•

1 Parent(s): f546c1a

update output

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ else:
 if st.button("Run"):
     with st.spinner("Working Hard..."):
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=src_lang, tgt_lang=tgt_lang)
-        trans_text = translator(sent)
         st.write(trans_text)
     st.success("Done!")

 if st.button("Run"):
     with st.spinner("Working Hard..."):
         translator = pipeline('translation', model=model, tokenizer=tokenizer, src_lang=src_lang, tgt_lang=tgt_lang)
+        trans_text = translator(sent)[0]["translation_text"]
         st.write(trans_text)
     st.success("Done!")