vives commited on
Commit
383394d
1 Parent(s): 8cc7e6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,8 +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_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'")
13
  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'")
 
 
14
  k = st.number_input("Top k nearest key-phrases",1,10,5)
15
 
16
  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 '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: