Update app.py

app.py

@@ -20,11 +20,12 @@ st.sidebar.write("""
 1. Write something in the text area (a prompt or random text) or use the dropdown menu to select predefined sample text.
 2. Select a task from the **task dropdown menu** below if you are providing your own text. **This is very important as it ensures the model responds accordingly.**
 3. If you are providing your own text, please do not select any predefined sample text from the dropdown menu.
-3. If a dropdown menu pops up for a nigerian language, **select the target nigerian language.**
+3. If a dropdown menu pops up for a nigerian language, **select the nigerian language (base language for diacritization and text cleaning, target language for translation).**
 4. Click the Generate button to get a response below the text area.\n
+5. For Translation tasks, setting english as the target language yields the best result.
 **Note: Model's overall performance vary (hallucinates) due to model size and training data distribution (majorly from news articles and the bible). Performance may worsen significantly with other task outside text generation.
 For other tasks, we suggest you try them several times due to the generator's sampling method.**\n
-5. Lastly, you can play with some of the generation parameters below to improve performance.
+6. Lastly, you can play with some of the generation parameters below to improve performance.
 """)
 
 max_length = 100 
@@ -159,12 +160,13 @@ instruction_wrap = {
 task_options = {
     "select": "{}",
     "Text Generation": "{}",
+    "Translation": "<translate> {} "
     "Sentiment Classification": "<classify> {} <sentiment>:",
     "Topic Classification": "<classify> {} <topic>",
     "Instruction Following" : "<prompt> {} <response>:",
     "Title Generation": "<title> {} <headline>",
-    "Text Diacritization": "<diacritize> {}",
-    "Text Cleaning": "<clean> {}"
+    "Text Diacritization": "<diacritize> {} ",
+    "Text Cleaning": "<clean> {} "
 }
 
 # Language options for diacritize and clean tasks
@@ -187,7 +189,7 @@ sample_text = st.selectbox("Select a sample text to test the model:", list(sampl
 task = st.selectbox("Select a task for the model:", list(task_options.keys()))
 
 # Conditionally show language options dropdown for diacritize and clean tasks
-if task in ["Diacritize text", "Clean text"]:
+if task in ["Diacritize text", "Clean text", "Translation"]:
     language = st.selectbox("Select a Nigerian language:", list(language_options.keys()))
     task_value = f"{task_options[task]} {language_options[language]}"
 else: