Update app.py

app.py

@@ -256,19 +256,20 @@ if st.button("Generate"):
             
             # generated_text  = re.split(r"\|(end_f_text|end_of_text|end_ofext|end_of_text_||end_of_te|end_oftext)|:|`", generated_text)[0]
             generated_text  = re.sub(r"\|(end_f_text|end_of_text|end_ofext|end_of_text_|end_of_te|end_o|end_of_tet|end_oftext)|:|`\|", "", generated_text)
+            generated_text = generated_text.strip("\n")
             print("Generated text: ", generated_text)
             
             if task == "Sentiment Classification" or ("Anyi na-echefu oke ike." in user_input and "Anyi na-echefu oke ike." in sample_texts.values()):
-                if generated_text.lower().startswith("negative"):
+                if generated_text.strip().lower().startswith("negative"):
                     generated_text = "Negative"
-                elif generated_text.lower().startswith("positive"):
+                elif generated_text.strip().lower().startswith("positive"):
                     generated_text = "Positive"
-                elif generated_text.lower().startswith("neutral"):
+                elif generated_text.strip().lower().startswith("neutral"):
                     generated_text = "Neutral"
     
             elif task == "Topic Classification" or ("Africa Free Trade Zone: Kò sí ìdènà láti kó ọjà láti orílẹ̀èdè kan sí òmíràn" in user_input and "Africa Free Trade Zone: Kò sí ìdènà láti kó ọjà láti orílẹ̀èdè kan sí òmíràn" in sample_texts.values()) :
                 generated_text  = generated_text[:15]
-                print("split", generated_text.split(" ")[0], generated_text.split(" "))
+                print("split", generated_text.split(" ")[0],  re.split(r"\.|\n|\*\*|\*", generated_text)[0], generated_text.split(" "))
                 generated_text = re.split(r"\.|\n|\*\*|\*", generated_text)[0]
                 
             elif task == "Translation":