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)
|