ajitrajasekharan commited on
Commit
90f04ee
1 Parent(s): 293e817

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -105,12 +105,8 @@ try:
105
  'Choose any of these sentences or type any text below',
106
  ('', "[MASK] who lives in New York and works for XCorp suffers from Parkinson's", "Lou Gehrig who lives in [MASK] and works for XCorp suffers from Parkinson's","'Lou Gehrig who lives in New York and works for [MASK] suffers from Parkinson's'","'Lou Gehrig who lives in New York and works for XCorp suffers from [MASK]'","[MASK] who lives in New York and works for XCorp suffers from Lou Gehrig's", "Parkinson who lives in [MASK] and works for XCorp suffers from Lou Gehrig's","Parkinson who lives in New York and works for [MASK] suffers from Lou Gehrig's","Parkinson who lives in New York and works for XCorp suffers from [MASK]","Lou Gehrig","Parkinson","Lou Gehrigh's is a [MASK]","Parkinson is a [MASK]","New York is a [MASK]","New York","XCorp","XCorp is a [MASK]","acute lymphoblastic leukemia","acute lymphoblastic leukemia is a [MASK]"))
107
  bert_tokenizer, bert_model = load_bert_model(model_name)
108
- default_text = "Imatinib is used to [MASK] acute lymphoblastic leukemia"
109
- input_text = st.text_area(
110
- label="Enter text below",
111
- value=default_text,
112
- )
113
- if st.button("Submit"):
114
  run_test(input_text,top_k)
115
  else:
116
  if len(option) > 0:
 
105
  'Choose any of these sentences or type any text below',
106
  ('', "[MASK] who lives in New York and works for XCorp suffers from Parkinson's", "Lou Gehrig who lives in [MASK] and works for XCorp suffers from Parkinson's","'Lou Gehrig who lives in New York and works for [MASK] suffers from Parkinson's'","'Lou Gehrig who lives in New York and works for XCorp suffers from [MASK]'","[MASK] who lives in New York and works for XCorp suffers from Lou Gehrig's", "Parkinson who lives in [MASK] and works for XCorp suffers from Lou Gehrig's","Parkinson who lives in New York and works for [MASK] suffers from Lou Gehrig's","Parkinson who lives in New York and works for XCorp suffers from [MASK]","Lou Gehrig","Parkinson","Lou Gehrigh's is a [MASK]","Parkinson is a [MASK]","New York is a [MASK]","New York","XCorp","XCorp is a [MASK]","acute lymphoblastic leukemia","acute lymphoblastic leukemia is a [MASK]"))
107
  bert_tokenizer, bert_model = load_bert_model(model_name)
108
+ input_text = st.text_input("Enter text below", "")
109
+ if len(input_text) > 0:
 
 
 
 
110
  run_test(input_text,top_k)
111
  else:
112
  if len(option) > 0: