mbahrami commited on
Commit
ac5b8a7
1 Parent(s): 78ff418

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -1,4 +1,3 @@
1
-
2
  import streamlit as st
3
 
4
  from transformers import pipeline
@@ -22,7 +21,7 @@ if text:
22
  data_load_state = st.text('Inference to model...')
23
  result = nlp(text+' '+nlp.tokenizer.mask_token)
24
  data_load_state.text('')
25
- for r in result:
26
  if r['sequence'].lower().strip() in history_keyword_text.lower().strip():
27
- r['score']*=100
28
  st.table(result)
 
 
1
  import streamlit as st
2
 
3
  from transformers import pipeline
 
21
  data_load_state = st.text('Inference to model...')
22
  result = nlp(text+' '+nlp.tokenizer.mask_token)
23
  data_load_state.text('')
24
+ for index, r in enumerate(result):
25
  if r['sequence'].lower().strip() in history_keyword_text.lower().strip():
26
+ result[index]['score']*=0.10
27
  st.table(result)