Abbas133 commited on
Commit
6daa488
·
verified ·
1 Parent(s): 2317cc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -8
app.py CHANGED
@@ -30,18 +30,19 @@ selected_language = st.selectbox("Select target language:", list(language_option
30
 
31
  if st.button("Translate"):
32
  if input_text:
33
- # Set target language
34
- target_language = language_options[selected_language]
35
- tokenizer.src_lang = "en"
36
- encoded_input = tokenizer(input_text, return_tensors="pt")
 
 
37
 
38
- # Generate translation
39
- generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(target_language))
40
- translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
41
 
42
  # Display translated text
43
  st.write(f"**Translated text ({selected_language}):**")
44
  st.write(translated_text)
45
  else:
46
  st.write("Please enter text to translate.")
47
-
 
30
 
31
  if st.button("Translate"):
32
  if input_text:
33
+ # Interactive loading state
34
+ with st.spinner('Translating... Please wait...'):
35
+ # Set target language
36
+ target_language = language_options[selected_language]
37
+ tokenizer.src_lang = "en"
38
+ encoded_input = tokenizer(input_text, return_tensors="pt")
39
 
40
+ # Generate translation
41
+ generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(target_language))
42
+ translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
43
 
44
  # Display translated text
45
  st.write(f"**Translated text ({selected_language}):**")
46
  st.write(translated_text)
47
  else:
48
  st.write("Please enter text to translate.")