Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import streamlit as st
 
 from transformers import pipeline
@@ -10,15 +9,16 @@ def get_model(model):
 
 text = st.text_input("Enter a text for auto completion...")
 
-history_keyword_text = st.text_input("Users's history keyword split with ','")
+history_keyword_text = st.text_input("Enter users's history keywords")
 
 model = st.selectbox("choose a model", ["roberta-base", "bert-base-uncased", "gpt2", "t5"])
 
 if text:
-	data_load_state = st.text('Loading...')
-	nlp = get_model(model)
-	result = nlp(text+nlp.tokenizer.mask_token)
-	data_load_state.text('Loading data...done!')
-#	for c in result:
-#		del c["token"]
-	st.table(result)
+    data_load_state = st.text('Loading...')
+    nlp = get_model(model)
+    result = nlp(text+nlp.tokenizer.mask_token)
+    data_load_state.text('Loading data...done!')
+    for c in result:
+        if c in history_keyword_text:
+            c['score']*=0.10
+    st.table(result)