PIERRE CUGNET commited on
Commit
3ba964c
1 Parent(s): 2847f6e

feat(py): add a nice sentence

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -87,7 +87,9 @@ prediction = model.predict({'input_ids' : encoded_input['input_ids'],'input_mask
87
  encoded_dict = {0: 'negative', 1: 'positive'}
88
 
89
  if np.argmax(prediction) == 0:
90
- st.write(f'Sentiment predicted : {encoded_dict[np.argmax(prediction)]}\nI\'m sorry you had a bad experience with our company :( , please accept our apologies')
 
91
  else:
92
- st.write(f'Sentiment predicted : {encoded_dict[np.argmax(prediction)]}\nGlad your flight was good ! Hope to see you soon :)')
 
93
 
 
87
  encoded_dict = {0: 'negative', 1: 'positive'}
88
 
89
  if np.argmax(prediction) == 0:
90
+ st.write(f'Sentiment predicted : {encoded_dict[np.argmax(prediction)]}')
91
+ st.write(f'I\'m sorry you had a bad experience with our company :( , please accept our apologies')
92
  else:
93
+ st.write(f'Sentiment predicted : {encoded_dict[np.argmax(prediction)]}\n')
94
+ st.write('Glad your flight was good ! Hope to see you soon :)')
95