Update app.py

app.py

@@ -3,9 +3,17 @@ import pandas as pd
 from transformers import MarianMTModel, MarianTokenizer
 
 # Load the model and tokenizer
-model_name = 'Helsinki-NLP/opus-mt-en-ur'
-model = MarianMTModel.from_pretrained(model_name)
-tokenizer = MarianTokenizer.from_pretrained(model_name)
+#model_name = 'Helsinki-NLP/opus-mt-en-ur'
+
+MODEL_NAME = os.environ.get("model_name")
+
+if not MODEL_NAME:
+    raise ValueError("MODEL_NAME is not set. Please add it in the Hugging Face Secrets.")
+
+
+
+model = MarianMTModel.from_pretrained(MODEL_NAME)
+tokenizer = MarianTokenizer.from_pretrained(MODEL_NAME)
 
 # Function to translate text from English to Urdu
 def translate_text(text):