nouman66 commited on
Commit
e4ce59a
·
1 Parent(s): 38651ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -9
app.py CHANGED
@@ -1,11 +1,7 @@
1
- pip install streamlit transformers
2
 
3
  import streamlit as st
4
  from transformers import pipeline
5
 
6
- # Rest of your Streamlit app code here
7
-
8
-
9
  def main():
10
  st.title("Multilingual Translator")
11
 
@@ -29,11 +25,9 @@ def get_languages():
29
  return ["en", "es", "fr", "de", "it", "ja", "ko", "zh-CN"]
30
 
31
  def translate_text(text, source_lang, target_lang):
32
- translator = Translator()
33
- translation = translator.translate(text, src=source_lang, dest=target_lang)
34
- return translation.text
35
 
36
  if __name__ == "__main__":
37
  main()
38
-
39
- streamlit run your_filename.py
 
 
1
 
2
  import streamlit as st
3
  from transformers import pipeline
4
 
 
 
 
5
  def main():
6
  st.title("Multilingual Translator")
7
 
 
25
  return ["en", "es", "fr", "de", "it", "ja", "ko", "zh-CN"]
26
 
27
  def translate_text(text, source_lang, target_lang):
28
+ translator = pipeline(task="translation", model=f"{source_lang}-to-{target_lang}")
29
+ translation = translator(text, max_length=500)[0]['translation_text']
30
+ return translation
31
 
32
  if __name__ == "__main__":
33
  main()