ajitrajasekharan
commited on
Commit
•
0feff30
1
Parent(s):
3af653e
Update app.py
Browse files
app.py
CHANGED
@@ -112,7 +112,7 @@ try:
|
|
112 |
'Choose any of these sentences or type any text below',
|
113 |
('', "[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]"))
|
114 |
input_text = st.text_input("Enter text below", "")
|
115 |
-
custom_model_name = st.text_input("Model not listed on left? Type the model name", "")
|
116 |
if (len(custom_model_name) > 0):
|
117 |
model_name = custom_model_name
|
118 |
bert_tokenizer, bert_model = load_bert_model(model_name)
|
|
|
112 |
'Choose any of these sentences or type any text below',
|
113 |
('', "[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]"))
|
114 |
input_text = st.text_input("Enter text below", "")
|
115 |
+
custom_model_name = st.text_input("Model not listed on left? Type the model name (fill-mask models only)", "")
|
116 |
if (len(custom_model_name) > 0):
|
117 |
model_name = custom_model_name
|
118 |
bert_tokenizer, bert_model = load_bert_model(model_name)
|