Update models/hsd_tr.py

models/hsd_tr.py

@@ -14,10 +14,11 @@ label_dict = {'LABEL_0': 'Hate', 'LABEL_1': 'Non-hate'}
 
 #user_input = st.text_input(" ")
 user_input = st.text_input("Enter your text here:",)
-if user_input:
+if st.button("Predict"):
+#if user_input:
     result = sentiment_pipeline(user_input)
     sentiment = result[0]["label"]
     label_dict = {'LABEL_0': 'Hate', 'LABEL_1': 'Non-hate'}
     sentiment = label_dict[sentiment]
-    #st.write(f"Detection: {sentiment}")
-    st.success(sentiment)
+    st.write(f"Detection: {sentiment}")
+    #st.success(sentiment)