Translate

app.py

@@ -1,15 +1,22 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
 
-print("Loading the model.......mm")
-print("Hello")
-st.title("Sentiment Analysis")
-st.write(" Powered by Hugging Face and Streamlit")
+st.title("Translation App (Kazakh to English)")
+st.write("Powered by Hugging Face and Streamlit")
 
+# Load model and tokenizer
+st.write("Loading the model, please wait...")
+model = AutoModelForSeq2SeqLM.from_pretrained('issai/tilmash')
+tokenizer = AutoTokenizer.from_pretrained("issai/tilmash")
 
-print('before slider')
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
-print("after slider")
+# Initialize translation pipeline
+tilmash = pipeline("translation", model=model, tokenizer=tokenizer, src_lang="kaz_Cyrl", tgt_lang="eng_Latn", max_length=1000)
 
- 
+# Input text
+input_text = st.text_area("Enter text in Kazakh:")
+if st.button("Translate"):
+    if input_text.strip():
+        translation = tilmash(input_text)
+        st.write("Translation:", translation[0]['translation_text'])
+    else:
+        st.write("Please enter some text to translate.")