Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
#
|
34 |
-
|
35 |
-
|
36 |
-
|
|
|
|
|
37 |
|
38 |
-
|
39 |
-
|
40 |
-
|
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.")
|
|