Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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:
|