Update app.py

app.py

@@ -7,18 +7,21 @@ def get_model(model):
 	return pipeline("fill-mask", model=model)
 
 
-text = st.text_input("Enter a text for auto completion...")
 
-history_keyword_text = st.text_input("Enter users's history keywords")
+
+history_keyword_text = st.text_input("Enter users's history keywords (optional, i.e., 'Gates')")
+
+text = st.text_input("Enter a text for auto completion...")
 
 model = st.selectbox("choose a model", ["roberta-base", "bert-base-uncased", "gpt2", "t5"])
 
 if text:
-    data_load_state = st.text('Loading...')
+    data_load_state = st.text('Loading model...')
     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
+    data_load_state = st.text('Inference to model...')
+    result = nlp(text+' '+nlp.tokenizer.mask_token)
+    data_load_state.text('')
+    for r in result:
+        if r in history_keyword_text:
+            r['score']*=0.10
     st.table(result)