vives commited on
Commit
121c34f
1 Parent(s): 383394d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,9 +9,9 @@ model_checkpoint = "vives/distilbert-base-uncased-finetuned-cvent-2019_2022"
9
  model = AutoModelForMaskedLM.from_pretrained(model_checkpoint, output_hidden_states=True)
10
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
11
  text = st.text_input("Enter word or key-phrase")
12
- exclude_words = st.radio("exclude_words",[True,False], help="Exclude results that contain any words in the query (i.e exclude 'hot coffee' if the query is 'coffee'")
13
 
14
- exclude_text = st.radio("exclude_text",[True,False], help="Exclude results that contain the query (i.e exclude 'tomato soup recipe' if the query is 'tomato soup'")
15
  k = st.number_input("Top k nearest key-phrases",1,10,5)
16
 
17
  with open("kp_dict_merged.pickle",'rb') as handle:
9
  model = AutoModelForMaskedLM.from_pretrained(model_checkpoint, output_hidden_states=True)
10
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
11
  text = st.text_input("Enter word or key-phrase")
12
+ exclude_words = st.radio("exclude_words",[True,False], help="Exclude results that contain any words in the query (i.e exclude 'hot coffee' if the query is 'cold coffee')")
13
 
14
+ exclude_text = st.radio("exclude_text",[True,False], help="Exclude results that contain the query (i.e exclude 'tomato soup recipe' if the query is 'tomato soup')")
15
  k = st.number_input("Top k nearest key-phrases",1,10,5)
16
 
17
  with open("kp_dict_merged.pickle",'rb') as handle: