Update app.py

app.py

@@ -24,10 +24,10 @@ device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 translator = None
 whisper_model = None
 
-def load_wolof_model():
+def load_darija_model():
     global translator
     if translator is None:
-        model_name = "LocaleNLP/eng_wolof"
+        model_name = "LocaleNLP/english_wolof"
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name).to(device)
         tokenizer = MarianTokenizer.from_pretrained(model_name)
         translator = pipeline("translation", model=model, tokenizer=tokenizer, device=0 if device.type == 'cuda' else -1)
@@ -91,7 +91,7 @@ def extract_text_from_file(uploaded_file):
             raise ValueError("Unsupported file type")
 
 def translate(text):
-    translator = load_wolof_model()
+    translator = load_darija_model()
     lang_tag = ">>wol<<"
 
     paragraphs = text.split("\n")
@@ -137,7 +137,7 @@ def translate_and_return(text):
 # Gradio UI components
 with gr.Blocks() as demo:
     gr.Markdown("## LocaleNLP English-to-Wolof Translator")
-    gr.Markdown("Upload English text, audio, or document to translate to Wolof using Localenlp Model.")
+    gr.Markdown("Upload English text, audio, or document to translate to Wolof using Localenlp model.")
 
     with gr.Row():
         input_mode = gr.Radio(choices=["Text", "Audio", "File"], label="Select input mode", value="Text")