update app.py

app.py

@@ -23,7 +23,7 @@ st.sidebar.write("""
 3. If a dropdown menu pops up for a nigerian language, **select the target nigerian language.**
 4. Click the Generate button to get a response below the text area.\n
 **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.**\n
+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.
 """)
 
@@ -128,13 +128,15 @@ async def generate_from_api(user_input, generation_config):
 # Sample texts
 sample_texts = {
     "select":"",
-    "Tell me a story in pidgin": "Tell me a story in Yoruba",
-    "Oma Ede, Mi ji ogede...": "Oma Ede, Mi ji ogede mi a foroma orhorho edha meji ri eka. ",
+    "Hausa: Afirka tana da al'adu...": "Afirka tana da al'adu da harsuna masu yawa. Tana da albarkatu da wuraren yawon shakatawa masu ban mamaki.",
+    "Yoruba: Ìmọ̀ sáyẹ́nsì àti...": "Ìmọ̀ sáyẹ́nsì àti tẹ̀knọ́lójì ń ṣe émi lóore tó níye lori ní Áfíríkà. Ó ń fún àwọn ènìyàn ní ànfààní láti dá irọyin àti kí wọ́n lè ṣe àwọn nǹkan tuntun.",
+    "Efik: Oma Ede, Mi ji ogede...": "Oma Ede, Mi ji ogede mi a foroma orhorho edha meji ri eka. ",
     "Igbo: N'ala Igbo ...": "N'ala Igbo, ọtụtụ ndị mmadụ kwenyere na e nwere mmiri ara na elu-ilu",
-    "who are you?": "who are you?",
     "urhobo: Eshare nana ri...":"Eshare nana ri vwo ẹguọnọ rẹ iyono rẹ Aristotle vẹ Plato na",
     "Efik: Ke eyo ...":"Ke eyo Jesus ye mme mbet esie, etop emi ama ada ifụre ọsọk mme Jew oro esịt okobụn̄ọde ke ntak idiọkido ke Israel, oro ẹkenyụn̄ ẹdude ke mfụhọ ke itie-ufụn mme nsunsu ido edinam Ido Ukpono Mme Jew eke akpa isua ikie.",
-    "Translate 'how are you?' to Yoruba": "Translate 'how are you?' to Yoruba",
+    "Tell me a story in pidgin": "Tell me a story Pidgin",
+    "who are you?": "who are you?",
+    # "Translate 'how are you?' to Yoruba": "Translate 'how are you?' to Yoruba",
     "Classify the sentiment": "Anyi na-echefu oke ike.",
     "what is the topic of this text": "Africa Free Trade Zone: Kò sí ìdènà láti kó ọjà láti orílẹ̀èdè kan sí òmíràn",
     "diacritize this text: ": "E sun, Alaga, fun ise amalayi ti e n se ni Naijiria. E maa ba a lo, egbon!", 
@@ -157,9 +159,9 @@ instruction_wrap = {
 task_options = {
     "select": "{}",
     "Text Generation": "{}",
-    "Instruction Following": "<prompt> {} <response>:",
     "Sentiment Classification": "<classify> {} <sentiment>:",
     "Topic Classification": "<classify> {} <topic>",
+    "Instruction Following" : "<prompt> {} <response>:",
     "Title Generation": "<title> {} <headline>",
     "Text Diacritization": "<diacritize> {}",
     "Text Cleaning": "<clean> {}"
@@ -199,7 +201,7 @@ def wrap_text(text, task_value):
 
 
 # Text input 
-user_input = st.text_area("Enter text below **(PLEASE, FIRST READ THE INSTRUCTIONS ON HOW TO USE IN THE SIDE BAR FOR BETTER EXPERIENCE)**: ", sample_texts[sample_text])
+user_input = st.text_area("Enter text below **(PLEASE, FIRST READ ALL INSTRUCTIONS IN THE SIDEBAR FOR A BETTER EXPERIENCE)**: ", sample_texts[sample_text])
 user_input = instruction_wrap.get(sample_texts.get(user_input, user_input), user_input)
 print("Final user input: ", user_input)
 if st.button("Generate"):
@@ -238,7 +240,7 @@ if st.button("Generate"):
                     generated_text = "Neutral"
 
             elif task == "Topic Classification":
-                generated_text  = generated_text[:20]
+                generated_text  = generated_text.split(" ")[0][:20]
                     
             full_output = st.empty()