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:
|