Update app.py

app.py

@@ -259,11 +259,11 @@ if st.button("Generate"):
             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 "negative" in generated_text.lower():
+                if generated_text.lower().startswith("negative"):
                     generated_text = "Negative"
-                elif "positive" in generated_text.lower():
+                elif generated_text.lower().startswith("positive"):
                     generated_text = "Positive"
-                elif "neutral" in generated_text.lower():
+                elif generated_text.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()) :