Spaces:
Build error
Build error
import streamlit as st | |
from transformers import MarianMTModel, MarianTokenizer | |
# Load pre-trained model and tokenizer | |
model_name = 'Helsinki-NLP/opus-mt-ur-de' | |
model = MarianMTModel.from_pretrained(model_name) | |
tokenizer = MarianTokenizer.from_pretrained(model_name) | |
# Function to translate text | |
def translate_text(text, src_lang, tgt_lang): | |
# Tokenize input text | |
inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True) | |
# Translate and decode | |
translated = model.generate(**inputs) | |
translated_text = tokenizer.decode(translated[0], skip_special_tokens=True) | |
return translated_text | |
# Streamlit app layout | |
st.title("Real-Time Urdu to German Translation") | |
st.write("Enter Urdu text below, and the app will translate it into German.") | |
# Input text area for Urdu text | |
input_text = st.text_area("Urdu Text", "", height=200) | |
# Translate when the button is pressed | |
if st.button("Translate"): | |
if input_text: | |
# Translate the text | |
translated_text = translate_text(input_text, "ur", "de") | |
st.subheader("Translated German Text:") | |
st.write(translated_text) | |
else: | |
st.write("Please enter some Urdu text to translate.") | |