Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
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)
|