shubhendu-ghosh commited on
Commit
8e92c05
1 Parent(s): 9c372b0

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -0
app.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import requests
3
+ import os
4
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
5
+ from transformers import pipeline
6
+
7
+ st.title("Translation App")
8
+
9
+ # Load the model and tokenizer
10
+ tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
11
+ model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
12
+
13
+ def translate(text, src_lang, tgt_lang):
14
+ translator = pipeline(
15
+ "translation",
16
+ model=model,
17
+ tokenizer=tokenizer,
18
+ src_lang=src_lang,
19
+ tgt_lang=tgt_lang,
20
+ )
21
+ output = translator(text, max_length=400)
22
+ return output[0]["translation_text"]
23
+
24
+ def main():
25
+ src_lang = st.text_input("Enter source language code (e.g., en):")
26
+ tgt_lang = st.text_input("Enter target language code (e.g., fr):")
27
+ text = st.text_area("Enter text to translate:")
28
+
29
+ if st.button("Translate"):
30
+ if src_lang and tgt_lang and text:
31
+ result = translate(text, src_lang, tgt_lang)
32
+ st.write("Translated Text:", result)
33
+ else:
34
+ st.warning("Please provide source language, target language, and text to translate.")
35
+
36
+ if __name__ == "__main__":
37
+ main()