witcher commited on
Commit
ebea8c9
·
verified ·
1 Parent(s): bbfb0f0

Upload 2 files

Browse files
Files changed (2) hide show
  1. main.py +25 -0
  2. requirements.txt +3 -0
main.py ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import Ncert.main as st
2
+ from transformers import pipeline
3
+
4
+ # Load translation pipeline
5
+ translation_pipe = pipeline("translation_en_to_awa", model="facebook/nllb-200-distilled-600M")
6
+
7
+ # Streamlit app
8
+ def main():
9
+ st.title("Text Translator")
10
+
11
+ # Input text box
12
+ input_text = st.text_area("Enter text to translate", "")
13
+
14
+ # Translate button
15
+ if st.button("Translate"):
16
+ if input_text.strip() != "":
17
+ translated_text = translation_pipe(input_text)[0]['translation_text']
18
+ st.write("Translated text:")
19
+ st.write(translated_text)
20
+ else:
21
+ st.warning("Please enter some text to translate.")
22
+
23
+
24
+ if __name__ == "__main__":
25
+ main()
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ transformers
2
+ streamlit
3
+ torch