Update app.py

app.py

@@ -153,13 +153,13 @@ def main():
             #prediction = loaded_model.predict(input_vector)[0]
             prediction_1 = round(proba_1[0])
             end_time = time.time()
+            st.header('Classic ML (LogReg on TF-IDF)')
             # Display the predicted sentiment
             if prediction_1 == 0:
                 st.write('The sentiment of your review is negative.')
                 st.write('Predicted probability:', (1 - round(proba_1[0], 2))*100, '%')
             else:
                 st.write('The sentiment of your review is positive.')
-                st.write('Predicted probability:', (round(proba_1[0], 2))*100, '%')
             st.write('Processing time:', round(end_time - start_time, 4), 'seconds')
 #         Lena
     if user_input is not None and submit:
@@ -169,6 +169,7 @@ def main():
             input_tokens = preprocess_text(user_input, 500, tokenizer)
             output = predict_sentiment(model2, input_tokens)
             end_time = time.time()
+            st.header('ErnieModel')
             st.write('The sentiment of your review is', output)
             st.write('Processing time:', round(end_time - start_time, 4), 'seconds')         
 # Gala 
@@ -178,6 +179,7 @@ def main():
             start_time = time.time()
             output = predict_sentence(user_input,model3)
             end_time = time.time()
+            st.header('bidirectional LSTM')
             st.write('The sentiment of your review is', output)
             st.write('Processing time:', round(end_time - start_time, 4), 'seconds')