Update app.py

app.py

@@ -6,6 +6,7 @@ from transformers import pipeline
 def get_model(model):
 	return pipeline("fill-mask", model=model, top_k=100)
 
+HISTORY_WEIGHT = 100 # set history weight (if found any keyword from history, it will priorities based on its weight)
 
 
 
@@ -25,8 +26,7 @@ if text:
     data_load_state.text('')
     for index, r in enumerate(result):
         if r['token_str'].lower().strip() in history_keyword_text.lower().strip():
-            st.caption(type(result))
-            result[index]['score']*=100
+            result[index]['score']*=HISTORY_WEIGHT
             
     result={k: v for k, v in sorted(result.items(), key=lambda item: item[0])}
     st.table(result)