Update app.py

app.py

@@ -1,13 +1,13 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
 
-# Load pre-trained model and tokenizer
-model_name = 'Helsinki-NLP/opus-mt-ur-de'
+# Load pre-trained multilingual model and tokenizer
+model_name = 'Helsinki-NLP/opus-mt-mul-de'  # Multilingual to German model
 model = MarianMTModel.from_pretrained(model_name)
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 
 # Function to translate text
-def translate_text(text, src_lang, tgt_lang):
+def translate_text(text):
     # Tokenize input text
     inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
 
@@ -27,7 +27,7 @@ input_text = st.text_area("Urdu Text", "", height=200)
 if st.button("Translate"):
     if input_text:
         # Translate the text
-        translated_text = translate_text(input_text, "ur", "de")
+        translated_text = translate_text(input_text)
         st.subheader("Translated German Text:")
         st.write(translated_text)
     else: