sarahai commited on
Commit
4dace66
·
verified ·
1 Parent(s): a18ecbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import streamlit as st
2
  from transformers import AutoModelForSeq2SeqLM, NllbTokenizer
3
 
4
- # Load model and tokenizer
5
  translation_model_name = 'sarahai/nllb-uzbek-cyrillic-to-russian'
6
  translation_model = AutoModelForSeq2SeqLM.from_pretrained(translation_model_name)
7
  translation_tokenizer = NllbTokenizer.from_pretrained(translation_model_name)
@@ -33,7 +32,7 @@ def translate(text, model, tokenizer, src_lang='uzb_Cyrl', tgt_lang='rus_Cyrl'):
33
  translated_chunks.append(tokenizer.decode(outputs[0], skip_special_tokens=True))
34
  return ' '.join(translated_chunks)
35
 
36
- # Custom CSS styles
37
  st.markdown("""
38
  <style>
39
  .big-font {
@@ -46,16 +45,16 @@ st.markdown("""
46
  </style>
47
  """, unsafe_allow_html=True)
48
 
49
- # Sidebar
50
  st.sidebar.markdown('## Навигация')
51
  uploaded_file = st.sidebar.file_uploader("Загрузите текстовый файл...", type=["txt"])
52
- process_btn = False # Define button state here
53
 
54
  if uploaded_file:
55
  st.sidebar.text("Файл загружен")
56
  process_btn = st.sidebar.button("Перевести")
57
 
58
- # Title and Description
59
  st.markdown('<h1 class="big-font">Перевод текста</h1>', unsafe_allow_html=True)
60
  st.markdown('<div class="big-font">Перевод с узбекского на русский</div>', unsafe_allow_html=True)
61
 
 
1
  import streamlit as st
2
  from transformers import AutoModelForSeq2SeqLM, NllbTokenizer
3
 
 
4
  translation_model_name = 'sarahai/nllb-uzbek-cyrillic-to-russian'
5
  translation_model = AutoModelForSeq2SeqLM.from_pretrained(translation_model_name)
6
  translation_tokenizer = NllbTokenizer.from_pretrained(translation_model_name)
 
32
  translated_chunks.append(tokenizer.decode(outputs[0], skip_special_tokens=True))
33
  return ' '.join(translated_chunks)
34
 
35
+
36
  st.markdown("""
37
  <style>
38
  .big-font {
 
45
  </style>
46
  """, unsafe_allow_html=True)
47
 
48
+
49
  st.sidebar.markdown('## Навигация')
50
  uploaded_file = st.sidebar.file_uploader("Загрузите текстовый файл...", type=["txt"])
51
+ process_btn = False
52
 
53
  if uploaded_file:
54
  st.sidebar.text("Файл загружен")
55
  process_btn = st.sidebar.button("Перевести")
56
 
57
+
58
  st.markdown('<h1 class="big-font">Перевод текста</h1>', unsafe_allow_html=True)
59
  st.markdown('<div class="big-font">Перевод с узбекского на русский</div>', unsafe_allow_html=True)
60